如何创建变量列表元组列表从选定的列在数据框？[重复]

Question

我有这样的df：

col1     col2      col3     col4
1         0         1        0 
1         0         1        0 
1         0         1        0 
1         0         1        0 
1         0         1        0 

如何将col1-col3转换为元组列表并分配给变量？然后创建一个只有col4的数组并赋值给变量？

示例最终结果：

lst = [[1,0,1],[1,0,1],[1,0,1],[1,0,1],[1,0,1]]]
lst_col4 = [[0],[0],[0],[0],[0]]

Answer 1

只需为这些colun编制索引，并使用.values属性：

lst = df[[col1, col2, col3]].values.tolist()

例子：

>>> df = pd.DataFrame({'a':[1,2,3], 'b':[4,5,6], 'c':[7,8,9]})
>>> lst = df[['a', 'c']].values.tolist()
>>> print(lst)
 [[1, 7], [2, 8],[3, 9]]