Python 将DataFrame复制到包含标题的列表列表
假设我有一个如下所示的数据帧:Python 将DataFrame复制到包含标题的列表列表,python,pandas,numpy,Python,Pandas,Numpy,假设我有一个如下所示的数据帧: 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
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
是否有一种方法可以将数据帧作为包含标题的列表列表返回
我可以返回如下列表中的标题和值
>>> 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]]
使用双转置:
或者更好:
您有两个列表,因此使用:
[df.columns.values.tolist()] + df.values.tolist()
>>> [['Col 1', 'Col 2', 'Col 3'], [1, 2, 3], [4, 5, 6], [7, 8, 9]]
很好,两个建议都比我最初的实现更有效。在我的第一次尝试中,我使用了
numpy.row\u stack()
,其实现与您用于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]]
[df.columns.values.tolist()] + df.values.tolist()
>>> [['Col 1', 'Col 2', 'Col 3'], [1, 2, 3], [4, 5, 6], [7, 8, 9]]