列名列表列表到 df 值列表列表

Question

给定以下表示列名的列表列表：

names = [['a','b'],['c','c'],['b','c']]

和以下数据框

df
   a  b  c
0  1  2  6
1  1  3  2
2  4  6  4

我想使用以下功能生成与names相同尺寸的列表：

lst = []
for idx, cols in enumerate(names):
  lst.append([])
  for col in cols:
    lst[-1].append(df.iloc[idx][col])

lst:
[[1,2],[2,2],[6,4]

即， names数组指向相关row_idx中从df中提取的列。

我试图避免嵌套循环。

Answer 1

您可以使用列表选择多个列

lst = []
for idx, cols in enumerate(names):
    lst.append(df.iloc[idx][cols].tolist())
# or list comprehension
lst = [df.iloc[idx][cols].tolist() for idx, cols in enumerate(names)]

print(lst)

[[1, 2], [2, 2], [6, 4]]

Answer 2

正如您所说， names的长度与数据帧的长度相同，并且您不想循环names或执行嵌套循环。 在那种情况下，是否允许在range循环？

index = range(len(names))

[df.iloc[i][names[i]].tolist() for i in index]
Out[16]: [[1, 2], [2, 2], [6, 4]]

或df.loc

[df.loc[i,names[i]].tolist() for i in index]
Out[35]: [[1, 2], [2, 2], [6, 4]]