python - 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]]

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]]

或更好 insert :

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]]