将熊猫数据框的 2 列转换为列表

Question

我知道您可以通过执行以下操作将单个列从数据框中拉出到列表中：

newList = df['column1'].tolist()

并且您可以将所有值转换为这样的列表：

newList = df.values.tolist()

但是有没有办法将 2 列从数据帧转换为列表，以便您得到一个看起来像这样的列表

    Column 1    Column 2 
0     apple      9
1     peach      12

结果列表是：

[[apple,9],[peach,12]]

谢谢

Answer 1

根据您的示例，您可以使用 df.values.tolist() 将 Pandas DataFrame 转换为列表。

如果你只想要特定的列，你只需要将这段代码中的 df 更改为只包含那些列的 df，如 df[[column1, column2, ..., columnN]].values.tolist()

Answer 2

您可以使用 zip：

[list(i) for i in zip(df['Column 1'], df['Column 2'])]

输出

[[apple,9],[peach,12]]

Answer 3

要将整个数据框转换为列表列表：

lst = df.to_numpy().tolist()