熊猫 - 选择列使用其他列值作为列名称

Question

我有一个数据框包含一列，我们称之为“名称”。 “名称”具有其他列的名称。我想添加一个新的列，每行都有一个基于该“名称”列中包含的列名称的值。

例子：

输入数据帧： pd.DataFrame.from_dict({"a": [1, 2, 3,4], "b": [-1,-2,-3,-4], "names":['a','b','a','b']})

 
    a | b | names | 
--- | --- | ---- | 
    1 | -1 | 'a' | 
    2 | -2 | 'b' | 
    3 | -3 | 'a' | 
    4 | -4 | 'b' | 

输出数据帧： pd.DataFrame.from_dict({"a": [1, 2, 3,4], "b": [-1,-2,-3,-4], "names":['a','b','a','b'], "new_col":[1,-2,3,-4]})

 
    a | b | names | new_col | 
--- | --- | ---- | ------ | 
    1 | -1 | 'a' | 1 | 
    2 | -2 | 'b' | -2 | 
    3 | -3 | 'a' | 3 | 
    4 | -4 | 'b' | -4 | 

谢谢

Answer 1

您可以使用lookup：

df['new_col'] = df.lookup(df.index, df.names) 
df 
# a b names new_col 
#0 1 -1  a 1 
#1 2 -2  b -2 
#2 3 -3  a 3 
#3 4 -4  b -4