解决方案1：使用apply

使用 pandas.DataFrame.apply 每行获取索引并获取相应的值

df['res'] = df.apply(lambda s: s[s['c']], axis=1)

输出：

a  b  c  res
0  0  3  a    0
1  1  4  a    1
2  2  5  b    5

解决方案2：直接切片

下面是一个更快的选择，直接切片底层numpy阵列：

# first map the columns to their position (a->0, b->1…)

cols = {k:v for v,k in enumerate(df.columns)}

# slice the elements

df['res'] = df.values[df.index, df['c'].map(cols)]

解决方案3

最后，一个有趣的选择 loc 及 numpy.diag 要获取对角线（警告：这是非常低效的内存）：

import numpy as np
df['res'] = np.diag(df.loc[df.index, df['c']])