如何将 Pandas DataFrame 列或行转换为列表
在 pandas DataFrame 中，每一列都由一个 pandas Series 对象表示。要获取列的列表表示形式，可以在 Series 对象上使用 tolist() 方法。例如：
<code class="python">import pandas as pd
data_dict = {'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 = pd.DataFrame(data_dict)
cluster_list = df['cluster'].tolist()
print(cluster_list)</code>输出：
['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C']
也可以直接使用list()将Series对象转换为列表：
<code class="python">cluster_list = list(df['cluster'])</code>
来获取整行的列表表示，您可以使用 DataFrame 的 iloc() 方法访问它。
<code class="python">row1_list = df.iloc[0].tolist() print(row1_list)</code>
输出：
[1000, '4000', 'Y']
同样，您可以转换整行直接到列表：
<code class="python">row1_list = list(df.iloc[0])</code>
以上是如何将 Pandas DataFrame 列或行转换为列表？的详细内容。更多信息请关注PHP中文网其他相关文章！
