如何使用来自指定列索引的值将列附加到数据框[重复]

Question

我有一个数据框，其中一列用于指定其他列中的哪一列具有我正在寻找的特定值。

df = pd.DataFrame({'COL1':['X','Y','Z'],'COL2':['A','B','C'],'X_SORT':['COL1','COL2','COL1']})

我正在尝试添加一个名为“X_SORT_VALUE”的新列并分配 X_SORT 列中标识的列的值。

df = df.assign(X_SORT_VALUE=lambda x: (x['X_SORT']))

但我得到了 X_SORT 列的值：

  COL1 COL2 X_SORT X_SORT_VALUE
0    X    A   COL1         COL1
1    Y    B   COL2         COL2
2    Z    C   COL1         COL1

而不是像我想要的那样获取该列索引的值：

  COL1 COL2 X_SORT X_SORT_VALUE
0    X    A   COL1         X
1    Y    B   COL2         B
2    Z    C   COL1         Z

Answer 1

你需要df.lookup这里：

df['X_SORT_VALUE'] = df.lookup(df.index,df['X_SORT'])

---

  COL1 COL2 X_SORT X_SORT_VALUE
0    X    A   COL1            X
1    Y    B   COL2            B
2    Z    C   COL1            Z