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

Question

我有一个包含几列的数据框，其中 'value1' 和 'value2'：

   value1  value2    other
0      13       8    xxxxx
1      14       7    xxxxx
2      17       7    xxxxx
3      18       7    xxxxx
4      19       8    xxxxx

目前我必须手动将 my_list 值编辑为以下格式的值对。

my_list = [[13,8],[14,7],[17,7],[18,7],[19,8]]

但是我已经能够将数据放入一个名为 value1 和 value2 的两列中的 pandas 数据框中，例如在我的 pandas value1 = 13 和 value2 = 8 等的第一行中。

有没有办法我可以编写类似 tolist() 的东西来输出上述结构中的值列表？

Answer 1

简短回答

使用底层numpy数组的tolist方法：df.values.tolist()

或df[['value1', 'value2']].values.tolist() 仅选择某些列

详情

您可以从列表转换为 DataFrame，反之亦然：

import pandas as pd

my_list = [[13,8],[14,7],[17,7],[18,7],[19,8]]

# list to dataframe
df = pd.DataFrame(my_list, columns=['value1', 'value2'])

# dataframe to list
my_list = df.values.tolist()

my_list:

[[13,8],[14,7],[17,7],[18,7],[19,8]]

df:

   value1  value2
0      13       8
1      14       7
2      17       7
3      18       7
4      19       8

如何仅将部分列转换为列表：

df[['value1', 'value2']].values.tolist()