python - 创建一个由其他列组合而成的列

Question

我有一个包含 3 列的 pandas 数据框，我想创建一个新列，如下所示:

import pandas as pd
import numpy as np
np.random.seed(2019)
df = pd.DataFrame(np.random.randn(5,3), columns=['A','B','C'])
cols = df.abs().idxmax(axis = 1)
cols
0    C
1    A
2    A
3    B
4    B
dtype: object

我想创建一个列 D，它采用索引 0 处的 C 的值、索引 1 处的 A 的值，等等。换句话说，D 根据对象 cols 从适当的列中获取值。我确信我可以循环索引并选择列，但是有更好的方法吗？

Answer 1

使用DataFrame.lookup :

print (df)
          A         B         C
0 -0.217679  0.821455  1.481278
1  1.331864 -0.361865  0.685609
2  0.573761  0.287728 -0.235634
3  0.953490 -1.689625 -0.344943
4  0.016905 -0.514984  0.244509

df['D'] = df.lookup(df.index, cols)
print (df)
          A         B         C         D
0 -0.217679  0.821455  1.481278  1.481278
1  1.331864 -0.361865  0.685609  1.331864
2  0.573761  0.287728 -0.235634  0.573761
3  0.953490 -1.689625 -0.344943 -1.689625
4  0.016905 -0.514984  0.244509 -0.514984