How to group dataframe rows into list in pandas groupby

Abhishek Thakur

I have a pandas data frame df like:

a b
A 1
A 2
B 5
B 5
B 4
C 6

I want to group by the first column and get second column as lists in rows:

A [1,2]
B [5,5,4]
C [6]

Is it possible to do something like this using pandas groupby?

EdChum

You can do this using groupby to group on the column of interest and then apply list to every group:

In [1]: df = pd.DataFrame( {'a':['A','A','B','B','B','C'], 'b':[1,2,5,5,4,6]})
        df

Out[1]: 
   a  b
0  A  1
1  A  2
2  B  5
3  B  5
4  B  4
5  C  6

In [2]: df.groupby('a')['b'].apply(list)
Out[2]: 
a
A       [1, 2]
B    [5, 5, 4]
C          [6]
Name: b, dtype: object

In [3]: df1 = df.groupby('a')['b'].apply(list).reset_index(name='new')
        df1
Out[3]: 
   a        new
0  A     [1, 2]
1  B  [5, 5, 4]
2  C        [6]

Collected from the Internet

Please contact [email protected] to delete if infringement.

edited at
0

Comments

0 comments
Login to comment

Related

From Dev

How can I get a list of values for rows combined by groupby in a Pandas Dataframe?

From Java

How to drop a list of rows from Pandas dataframe?

From Dev

How to select a list of rows by name in Pandas dataframe?

From Java

Efficient way to group pandas dataframe rows by a list of tags in a column

From Dev

Pandas group by on groupby to list of lists

From Dev

How to filter a pandas dataframe to just the rows that show change within a group?

From Dev

Drop pandas dataframe rows based on groupby() condition

From Dev

using groupby on pandas dataframe to group by financial year

From Dev

Pandas Dataframe GroupBy - Displaying Group Statistics

From Dev

using groupby on pandas dataframe to group by financial year

From Dev

Converting groupby dataframe (with dropped duplicate rows by group) into normal dataframe

From Dev

How to generate a list of ranges to subset rows from pandas dataframe?

From Dev

Pandas: How to remove rows from a dataframe based on a list?

From Java

Pandas dataframe groupby make a list or array of a column

From Dev

Groupby items within a list in a Pandas DataFrame

From Dev

How do I get the first timestamp (index) of a group when applying groupby to a python pandas dataframe?

From Java

Group by and aggregate rows into list of series or dicts in Pandas

From Dev

Python Pandas, slice rows from group in .groupby().apply()

From Dev

How to broadcast Pandas groupby result to all rows?

From Dev

How to get last group in Pandas' groupBy?

From Dev

Pandas: how to get a particular group after groupby?

From Dev

Group rows in dataframe by assigning values as a column in pandas dataframe

From Java

Update rows in Pandas Dataframe based on the list values

From Dev

Transform rows of pandas Dataframe into list of strings

From Dev

Selecting rows from pandas DataFrame using a list

From Dev

List Comprehension Over Pandas Dataframe Rows

From Dev

Insert rows in Pandas dataframe after comparing with list

From Dev

Inserting named rows in dataframe based on a list pandas

From Dev

Split list in rows python pandas dataframe

Related Related

  1. 1

    How can I get a list of values for rows combined by groupby in a Pandas Dataframe?

  2. 2

    How to drop a list of rows from Pandas dataframe?

  3. 3

    How to select a list of rows by name in Pandas dataframe?

  4. 4

    Efficient way to group pandas dataframe rows by a list of tags in a column

  5. 5

    Pandas group by on groupby to list of lists

  6. 6

    How to filter a pandas dataframe to just the rows that show change within a group?

  7. 7

    Drop pandas dataframe rows based on groupby() condition

  8. 8

    using groupby on pandas dataframe to group by financial year

  9. 9

    Pandas Dataframe GroupBy - Displaying Group Statistics

  10. 10

    using groupby on pandas dataframe to group by financial year

  11. 11

    Converting groupby dataframe (with dropped duplicate rows by group) into normal dataframe

  12. 12

    How to generate a list of ranges to subset rows from pandas dataframe?

  13. 13

    Pandas: How to remove rows from a dataframe based on a list?

  14. 14

    Pandas dataframe groupby make a list or array of a column

  15. 15

    Groupby items within a list in a Pandas DataFrame

  16. 16

    How do I get the first timestamp (index) of a group when applying groupby to a python pandas dataframe?

  17. 17

    Group by and aggregate rows into list of series or dicts in Pandas

  18. 18

    Python Pandas, slice rows from group in .groupby().apply()

  19. 19

    How to broadcast Pandas groupby result to all rows?

  20. 20

    How to get last group in Pandas' groupBy?

  21. 21

    Pandas: how to get a particular group after groupby?

  22. 22

    Group rows in dataframe by assigning values as a column in pandas dataframe

  23. 23

    Update rows in Pandas Dataframe based on the list values

  24. 24

    Transform rows of pandas Dataframe into list of strings

  25. 25

    Selecting rows from pandas DataFrame using a list

  26. 26

    List Comprehension Over Pandas Dataframe Rows

  27. 27

    Insert rows in Pandas dataframe after comparing with list

  28. 28

    Inserting named rows in dataframe based on a list pandas

  29. 29

    Split list in rows python pandas dataframe

HotTag

Archive