We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 143f657 commit ec34173Copy full SHA for ec34173
pandas/tests/groupby/aggregate/test_other.py
@@ -512,3 +512,14 @@ def test_agg_category_nansum(observed):
512
if observed:
513
expected = expected[expected != 0]
514
tm.assert_series_equal(result, expected)
515
+
516
517
+def test_agg_list_like_func():
518
+ # GH 18473
519
+ df = pd.DataFrame({'A': [str(x) for x in range(3)],
520
+ 'B': [str(x) for x in range(3)]})
521
+ grouped = df.groupby('A', as_index=False, sort=False)
522
+ result = grouped.agg({'B': lambda x: list(x)})
523
+ expected = pd.DataFrame({'A': [str(x) for x in range(3)],
524
+ 'B': [[str(x)] for x in range(3)]})
525
+ tm.assert_frame_equal(result, expected)
0 commit comments