如何从 Pandas DataFrames 中提取数据作为列表？

从 Pandas DataFrame 获取列表

使用 Pandas DataFrame 时，通常需要将数据提取为列表以进行进一步分析或处理。以下是您问题的答案：

以列表形式检索列

要以列表形式检索列，只需将其作为 Pandas 系列访问，然后进行转换使用 .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()

这将返回一个包含“簇”列中的值的列表：

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

检索行或列作为列表

要以列表形式检索行或列的全部内容，请使用 .values 属性。

例如，要以列表形式检索第 2 行：

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

并将第 3 列作为列表检索：

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

以上是如何从 Pandas DataFrames 中提取数据作为列表？的详细内容。更多信息请关注PHP中文网其他相关文章！