@@ -254,14 +254,15 @@ def test_groupby_quantile_NA_float(any_float_allowed_nullable_dtype):
254
254
{"x" : [1 , 1 ], "y" : [0.2 , np .nan ]}, dtype = any_float_allowed_nullable_dtype
255
255
)
256
256
result = df .groupby ("x" )["y" ].quantile (0.5 )
257
- expected = pd .Series ([0.2 ], dtype = float , index = [1.0 ], name = "y" )
258
- expected .index .name = "x"
257
+ expected = pd .Series ([0.2 ], dtype = float , index = Index (df ["x" ][:1 ]), name = "y" )
259
258
tm .assert_series_equal (expected , result )
260
259
261
260
result = df .groupby ("x" )["y" ].quantile ([0.5 , 0.75 ])
262
261
expected = pd .Series (
263
262
[0.2 ] * 2 ,
264
- index = pd .MultiIndex .from_product (([1.0 ], [0.5 , 0.75 ]), names = ["x" , None ]),
263
+ index = pd .MultiIndex .from_arrays (
264
+ [Index (df ["x" ]), [0.5 , 0.75 ]], names = ["x" , None ]
265
+ ),
265
266
name = "y" ,
266
267
)
267
268
tm .assert_series_equal (result , expected )
@@ -271,11 +272,11 @@ def test_groupby_quantile_NA_int(any_nullable_int_dtype):
271
272
# GH#42849
272
273
df = DataFrame ({"x" : [1 , 1 ], "y" : [2 , 5 ]}, dtype = any_nullable_int_dtype )
273
274
result = df .groupby ("x" )["y" ].quantile (0.5 )
274
- expected = pd .Series ([3.5 ], dtype = float , index = Index ([ 1 ], name = "x" ), name = "y" )
275
+ expected = pd .Series ([3.5 ], dtype = float , index = Index (df [ "x" ][: 1 ] ), name = "y" )
275
276
tm .assert_series_equal (expected , result )
276
277
277
278
result = df .groupby ("x" ).quantile (0.5 )
278
- expected = DataFrame ({"y" : 3.5 }, index = Index ([ 1 ], name = "x" ))
279
+ expected = DataFrame ({"y" : 3.5 }, index = Index (df [ "x" ][: 1 ] ))
279
280
tm .assert_frame_equal (result , expected )
280
281
281
282
@@ -284,8 +285,7 @@ def test_groupby_quantile_allNA_column(dtype):
284
285
# GH#42849
285
286
df = DataFrame ({"x" : [1 , 1 ], "y" : [pd .NA ] * 2 }, dtype = dtype )
286
287
result = df .groupby ("x" )["y" ].quantile (0.5 )
287
- expected = pd .Series ([np .nan ], dtype = float , index = [1.0 ], name = "y" )
288
- expected .index .name = "x"
288
+ expected = pd .Series ([np .nan ], dtype = float , index = Index (df ["x" ][:1 ]), name = "y" )
289
289
tm .assert_series_equal (expected , result )
290
290
291
291
0 commit comments