将带有列名和索引名的 python pandas dataframe 转换为列表列表

Question

我想像下面那样转换 dataframe（其中 ix 是数据帧的索引）

九	一种	乙
X	0	0
是	0	0

像这样的列表列表：

[ [ix, A, B], [X, 0, 0], [Y, 0, 0] ]

创建一个仍然包含 dataframe 列和索引名称 A、B 和 X、Y 的矩阵（列表列表）。

Answer 1

将columns转换为列表，将 append 转换为body列表：

[df.columns.tolist()] + df.values.tolist()

[['ix', 'A', 'B'], ['X', 0, 0], ['Y', 0, 0]]

或者，如果您只想将其存储在某处以实现可重复性，您可能需要使用to_dict方法：

df.to_dict('list')
{'ix': ['X', 'Y'], 'A': [0, 0], 'B': [0, 0]}

Answer 2

如果你所有的列都是相同的类型，我会使用.to_numpy然后使用类似的东西将列添加到数组中：

a = df.to_numpy()
a = numpy.vstack([df.columns, a])

这涉及更多的进口（技术上），但你已经在进口 pandas 所以 numpy 没有什么可以引进的。