在另一列中列出的pandas中提取特定列名的值

Question

标题不太清楚，但这是一个例子。假设我有：

person  apple  orange  type
Alice   11     23      apple
Bob     14     20      orange

我希望得到这个专栏

person new_col
Alice  11
Bob    20

所以我们得到行'Alice'的'apple'列和'Bob'行的'orange'。

我在想它，但那会很慢。有更快的方法吗？

Answer 1

使用DataFrame.lookup：

df['new_col'] = df.lookup(df.index, df['type'])
print (df)
  person  apple  orange    type  new_col
0  Alice     11      23   apple       11
1    Bob     14      20  orange       20

如果只想使用2列DataFrame，请使用assign或DataFrame构造函数：

df1 = df[['person']].assign(new_col=df.lookup(df.index, df['type']))
print (df1)
  person  new_col
0  Alice       11
1    Bob       20

df1 = pd.DataFrame({
        'person':df['person'].values,
        'new_col':df.lookup(df.index, df['type'])},
         columns=['person','new_col'])
print (df1)
  person  new_col
0  Alice       11
1    Bob       20