python pandas将数据框展平为列表

Question

我有一个像这样的 df:

import pandas
a=[['1/2/2014', 'a', '6', 'z1'], 
   ['1/2/2014', 'a', '3', 'z1'], 
   ['1/3/2014', 'c', '1', 'x3'],
   ]
df = pandas.DataFrame.from_records(a[1:],columns=a[0])

我想把 df 展平，所以它是一个像这样的连续列表:

['1/2/2014', 'a', '6', 'z1', '1/2/2014', 'a', '3', 'z1','1/3/2014', 'c', '1', 'x3']

我可以遍历行并将 extend 到一个列表，但这样做更简单吗？

Answer 1

您可以使用 .flatten()在 DataFrame 上转换为 NumPy 数组:

df.to_numpy().flatten()

您也可以添加.tolist()如果您希望结果是 Python list。

编辑

在以前的 Pandas 版本中，使用 values 属性而不是 .to_numpy() 方法，如下面的评论中所述。