पायथन - पैंडास में डेटाफ्रेम पंक्तियों को सूची में कैसे समूहित करें?

डेटाफ्रेम पंक्तियों को सूची में समूहित करने के लिए, apply() फ़ंक्शन का उपयोग करें। सबसे पहले, आवश्यक लाइब्रेरी आयात करें:

import pandas as pd

2 कॉलम के साथ डेटाफ्रेम बनाएं:

dataFrame = pd.DataFrame(
    {
        "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
        "Units": [100, 150, 110, 80, 110, 90]
    }
)

apply(list) के साथ डेटाफ्रेम को सूची में समूहित करना:

dataFrame = dataFrame.groupby('Car')['Units'].apply(list)

उदाहरण

निम्नलिखित कोड है:

import pandas as pd
# Create DataFrame
dataFrame = pd.DataFrame(
    {
        "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
        "Units": [100, 150, 110, 80, 110, 90]
    }
)
print("DataFrame ...\n", dataFrame)
# grouping dataframe into lists
dataFrame = dataFrame.groupby('Car')['Units'].apply(list)
# displaying dataframe lists
print("\nDataFrame ...\n", dataFrame)

आउटपुट

यह निम्नलिखित आउटपुट देगा:

DataFrame ... 
    Car  Units
0  BMW    100
1  Lexus  150
2  Audi   110
3  Mustang 80
4  Bentley 110
5  Jaguar 90
DataFrame ... 
Car
Audi     [110]
BMW      [100]
Bentley  [110]
Jaguar   [90]
Lexus    [150]
Mustang  [80]
Name: Units, dtype: object