Skip to content
This repository was archived by the owner on Feb 2, 2024. It is now read-only.

Re-implement df.head based on new structure #855

Merged
densmirn merged 3 commits into from
May 25, 2020
Merged

Re-implement df.head based on new structure #855

Changes from 1 commit
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
6db8e0c
Re-implement df.head based on new structure
densmirn May 21, 2020
02e281a
Merge branch 'feature/dataframe_model_refactoring' of https://github.…
densmirn May 25, 2020
e13213a
Simplify codegen docstring
densmirn May 25, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Simplify codegen docstring
  • Loading branch information
@densmirn
densmirn authored May 25, 2020
commit e13213a20bbf5895eae3fa723362f560c688da8a
8 changes: 2 additions & 6 deletions sdc/datatypes/hpat_pandas_dataframe_functions.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -454,19 +454,15 @@ def sdc_pandas_dataframe_reduce_columns(df, func_name, params, ser_params):

def _dataframe_reduce_columns_codegen_head(func_name, func_params, series_params, df):
"""
Example func_text for func_name='head' columns=('float', 'int', 'string'):
Example func_text for func_name='head' columns=('float', 'string'):
def _df_head_impl(df, n=5):
data_0 = df._data[0][0]
series_0 = pandas.Series(data_0)
result_0 = series_0.head(n=n)
data_1 = df._data[1][0]
series_1 = pandas.Series(data_1)
result_1 = series_1.head(n=n)
data_2 = df._data[2][0]
series_2 = pandas.Series(data_2)
result_2 = series_2.head(n=n)
return pandas.DataFrame({"float": result_0, "int": result_1, "string": result_2},
index=df._index[:n])
return pandas.DataFrame({"float": result_0, "string": result_1}, index=df._index[:n])
"""
results = []
joined = ', '.join(func_params)
Expand Down