File tree Expand file tree Collapse file tree 1 file changed +22
-0
lines changed Expand file tree Collapse file tree 1 file changed +22
-0
lines changed Original file line number Diff line number Diff line change @@ -2003,3 +2003,25 @@ def test_sum_with_level():
20032003 }
20042004 modin_df , pandas_df = pd .DataFrame (data ), pandas .DataFrame (data )
20052005 eval_general (modin_df , pandas_df , lambda df : df .set_index ("C" ).groupby ("C" ).sum ())
2006+
2007+
2008+ def test_reset_index_groupby ():
2009+ frame_data = np .random .randint (97 , 198 , size = (2 ** 6 , 2 ** 4 ))
2010+ pandas_df = pandas .DataFrame (
2011+ frame_data ,
2012+ index = pandas .MultiIndex .from_tuples (
2013+ [(i // 4 , i // 2 , i ) for i in range (2 ** 6 )]
2014+ ),
2015+ ).add_prefix ("col" )
2016+ pandas_df .index .names = [f"index_{ i } " for i in range (len (pandas_df .index .names ))]
2017+ # Convert every other column to string
2018+ for col in pandas_df .iloc [
2019+ :, [i for i in range (len (pandas_df .columns )) if i % 2 == 0 ]
2020+ ]:
2021+ pandas_df [col ] = [str (chr (i )) for i in pandas_df [col ]]
2022+ modin_df = from_pandas (pandas_df )
2023+ eval_general (
2024+ modin_df ,
2025+ pandas_df ,
2026+ lambda df : df .reset_index ().groupby (["index_0" , "index_1" ]).count (),
2027+ )
You can’t perform that action at this time.
0 commit comments