首页 > 解决方案 > 如何将 Python 数据框转换为列表列表?

问题描述

我有一个尺寸为 21392x1972 的 Python 数据框。我想要完成的是将数据框转换为列表列表,这样我的数据框的第一列是长列表中的第一个列表,数据框的第二列是带有一个长列表的第二个列表和很快。

我尝试使用 tolist() 将数据框转换为列表列表。现在发生的事情是我的数据框的每一行都成为长列表中的一个列表。但是,我想要完成的是数据框的每一列都应该成为长列表中的一个列表。我是使用 Pandas 和 Python 的新手,因此非常感谢这方面的任何帮助。干杯!

import pandas as pd
mydataset = pd.read_csv('final_merged_data.csv')
mydataset_seq = mydataset.values.tolist() 

标签: pythonpandascasting

解决方案


遍历数据框中的所有列,索引该列,并将其转换为列表:

lst = [df[i].tolist() for i in df.columns]

例子:

df = pd.DataFrame({'a' : [1, 2, 3, 4],
'b' : [5, 6, 7, 8]})

print(df)
print('list', [df[i].tolist() for i in df.columns])

输出:

   a  b
0  1  5
1  2  6
2  3  7
3  4  8
'list' [[1, 2, 3, 4], [5, 6, 7, 8]]

推荐阅读