How to Extract Data from Pandas DataFrames as Lists?

Getting Lists from Pandas DataFrames

When working with Pandas DataFrames, it's often necessary to extract data as lists for further analysis or processing. Here are the answers to your questions:

Retrieving a Column as a List

To retrieve a column as a list, simply access it as a Pandas Series and then convert it to a list using .tolist().

import pandas as pd

df = pd.DataFrame({
    'cluster': ['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'],
    'load_date': ['1/1/2014', '2/1/2014', '3/1/2014', '4/1/2014', '4/1/2014', '4/1/2014', '7/1/2014', '8/1/2014', '9/1/2014'],
    'budget': [1000, 12000, 36000, 15000, 12000, 90000, 22000, 30000, 53000],
    'actual': [4000, 10000, 2000, 10000, 11500, 11000, 18000, 28960, 51200],
    'fixed_price': ['Y', 'Y', 'Y', 'N', 'N', 'N', 'N', 'N', 'N']
})

df['cluster'].tolist()

This will return a list containing the values in the 'cluster' column:

['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C']

Retrieving a Row or Column as a List

To retrieve the entire contents of a row or column as a list, use the .values attribute.

For example, to retrieve row 2 as a list:

df.iloc[1].values.tolist()

And to retrieve column 3 as a list:

df['actual'].values.tolist()

The above is the detailed content of How to Extract Data from Pandas DataFrames as Lists?. For more information, please follow other related articles on the PHP Chinese website!