如何获取列中的数据，名称保存在 python 中的另一列中

Question

1. 我有一个数据框：

   import pandas as pd
   data = {'Alpha': [1, 2, 3, 4, 5], 'Beta':[6, 7 ,8, 9, 10],
   'Delta': [11, 12, 13, 14, 15], 'Gamma': [16, 17, 18, 19, 20]}
   df = pd.DataFrame(data, index = ['2010', '2011', '2012', '2013', '2014'])
   df
   

2. 它还有一列'Select'，包含其他列名

   df['Select'] = ['Beta', 'Gamma', 'Gamma', 'Delta', 'Alpha']
   df
   

3. 我需要添加一列，比如 'Choice'，其中数据将从列中派生，其名称保存在 'Select' 列中，如下所示：

   df['Choice'] = [0,0,0,0,0]
   row = df['Select'].values
   for i in range(len(row)):
       df['Choice'][i] = df[row[i]][i]
   df
   

4. 我是用循环来做的，但也许有更优雅的方法？谢谢！

Answer 1

我们可以使用 DataFrame.lookup():

In [49]: df['new'] = df.lookup(df.index, df.Select)

In [50]: df
Out[50]:
      Alpha  Beta  Delta  Gamma Select  Choice  new
2010      1     6     11     16   Beta       6    6
2011      2     7     12     17  Gamma      17   17
2012      3     8     13     18  Gamma      18   18
2013      4     9     14     19  Delta      14   14
2014      5    10     15     20  Alpha       5    5