How to create variable list of list of tuples from selected columns in dataframe?
本问题已经有最佳答案,请猛点这里访问。
我有这样的df:
1 2 3 4 5 6 | col1 col2 col3 col4 1 0 1 0 1 0 1 0 1 0 1 0 1 0 1 0 1 0 1 0 |
如何将
最终结果示例:
1 2 | lst = [[1,0,1],[1,0,1],[1,0,1],[1,0,1],[1,0,1]]] lst_col4 = [[0],[0],[0],[0],[0]] |
只需索引这些列,并使用
1 | lst = df[[col1, col2, col3]].values.tolist() |
例子:
1 2 3 4 | >>> 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]] |