如何从数据框中的选定列创建元组列表的变量列表？

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

只需索引这些颜色，然后使用.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]]