Python Pandas：从列分配中调用列名

Question

我有以下数据框：number_1 number_2 number_3 ... number_100 index1        2        2000         8          3201      10       21           2          1...我想创建一个新列，它等于number_index这样：number_1 number_2 number_3 ... number_100 index number_index1        2        2000         8          3     2000201      10       21           2          1     201...有table['number_index'] = table['number_' + str(table['index'])]可能吗？

Answer 1

你快到了。由于您必须为每一行都做，因此您可以使用lambda函数来完成这项工作。

table['number_index'] = table.apply(lambda x: x["number_" + str(x['index'])], axis = 1)

Answer 2

在名称lookup之后使用str splitcolumns

df1=df.copy()

df1.columns=df1.columns.str.split('_').str[-1]

df['Newval']=df1.lookup(df.index,df['index'].astype(str))

df

Out[34]: 

   number_1  number_2  number_3  number_100  index  Newval

0         1         2      2000           8      3    2000

1       201        10        21           2      1     201

Answer 3

我真的不明白你的问题。如果您想在名称中放置一个变量，我会建议使用该exec命令。

exec("table['number_index'] = " + f'''table['number_' + {str(table['index'])}]'''

因此，您可以执行字符串命令。随着f'''{var_name}'''您可以将字符串和变量。