Python Pandas : Call column name from column assignment

Question

我有以下数据框:

number_1 number_2 number_3 ... number_100 index
1        2        2000         8          3
201      10       21           2          1
...

我想创建一个新列，它等于 number_index，如下所示:

number_1 number_2 number_3 ... number_100 index number_index
1        2        2000         8          3     2000
201      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)