python - 数据帧 : looping through calling some variables while manually calling others

Question

我有一个数据框data，它包含多个名为:

 y        E4        C1  ...  season_2  season_3      ar1

我有列表collistful，它包含数据帧数据中一些变量的各种组合:

['E1'],
 ['C1'],
 ['C2'],
 ['C3'],
 ['C4'],
 ['G1'],
 ['E1', 'E2'],
 ['E1', 'E3'],
 ['E1', 'E4'],
 ['E1', 'C1'],
 ['E1', 'C2'],
 ['E1', 'C3'],
 ['E1', 'C4'],
 ['E1', 'G1'],
 ['E2', 'E3'],
 ['E2', 'E4'],
 ['E2', 'C1'],
 ['E2', 'C2'],
 ['E2', 'C3'],
 ['E2', 'C4'],
 ['E2', 'G1'],

我想做的是编写一个循环，调用/循环 collistfull 中的变量，同时还手动调用每个循环中包含的某些变量，例如那:

for col in collistfull: 
    datanew = data[(col),"y","season_2","ar1"]
    print(datadrop1)

但是我上面尝试的代码返回错误:TypeError: '(['E1'], 'y', 'season_2', 'ar1')' is an invalid key

Answer 1

您可以简单地执行以下操作:

for col in collistfull: 
    datanew = data[col + ["y","season_2","ar1"]]
    print(datanew)

由于每个 col 都是一个列表，我们可以简单地使用 + 运算符将新名称附加到 col 中。