将 DataFrame 转换为具有 DataFrame 列名的多维数组答案

【问题标题】：Convert DataFrame into multi-dimensional array with the column names of DataFrame将 DataFrame 转换为具有 DataFrame 列名的多维数组
【发布时间】：2019-07-22 09:58:37
【问题描述】：

下面是我要操作的 DataFrame：

df = pd.DataFrame({'A': [1,1,1],
                   'B': [2,2,3],
                   'C': [4,5,4]})

df 的每一行都会创建一个唯一键。目标是创建以下多维数组列表：

parameter = [[['A', 1],['B', 2], ['C', 4]],
             [['A', 1],['B', 2], ['C', 5]],
             [['A', 1],['B', 3], ['C', 4]]]

问题与this question 相关，我必须在parameter 上进行迭代，但我必须将df（行）中的所有parameter 放入列表中，而不是手动将它们提供给我的函数。

【问题讨论】：

标签： python list dataframe multidimensional-array

【解决方案1】：

您可以使用以下列表推导，它将每行上的值与数据框的列一起压缩：

from itertools import repeat

[list(map(list,zip(cols, i))) for cols, i in zip(df.values.tolist(), repeat(df.columns))]

[[[1, 'A'], [2, 'B'], [4, 'C']],
 [[1, 'A'], [2, 'B'], [5, 'C']],
 [[1, 'A'], [3, 'B'], [4, 'C']]]

【讨论】：