Pandas DataFrame转换为包括表头的列表嵌套列表

Question

假设我有一个类似这样的DataFrame：

df = pd.DataFrame([[1, 2, 3], 
                   [4, 5, 6], 
                   [7, 8, 9]], 
                   columns=['Col 1', 'Col 2', 'Col 3'])
>>> df
   Col 1  Col 2  Col 3
0      1      2      3
1      4      5      6
2      7      8      9

有没有一种Pandas的方法可以返回一个包含标题的列表列表的DataFrame？

我可以按以下方式返回标题和值的列表：

>>> df.columns.values.tolist()
['Col 1', 'Col 2', 'Col 3']
>>> df.values.tolist()
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
>>> df.tolist()

但是我该如何返回以下结果？

[['Col 1', 'Col 2', 'Col 3'], [1, 2, 3], [4, 5, 6], [7, 8, 9]]

- ScottMcC

Answer 1

使用双重转置和reset_index：

print (df.T.reset_index().values.T.tolist())
[['Col 1', 'Col 2', 'Col 3'], [1, 2, 3], [4, 5, 6], [7, 8, 9]]

或者更好的插入：

a = df.columns.values.tolist()
b = df.values.tolist()

b.insert(0, a)
print (b)
[['Col 1', 'Col 2', 'Col 3'], [1, 2, 3], [4, 5, 6], [7, 8, 9]]

- jezrael

---

很好，两个建议都比我的原始实现更好。对于我的第一个尝试，我使用了numpy.row_stack()，与您用于insert的实现类似。 - ScottMcC

Answer 2

您有两个列表，因此可以使用：

[df.columns.values.tolist()] + df.values.tolist()
>>> [['Col 1', 'Col 2', 'Col 3'], [1, 2, 3], [4, 5, 6], [7, 8, 9]]

- patricio