接入列动态地基于价值的其他列大熊猫

Question

比方说，我有以下的数据帧：

idx | X1 | X2 | X3 | Z 
----------------------- 
    1 | 5 | 8 | 4 | 1 
    2 | 3 | 2 | 6 | 2 
    3 | 6 | 3 | 2 | 1 

对于每一行，我现在想访问由值指示的列Z，即我想要的结果是：

idx | X 
-------- 
    1 | 5 # from X1 
    2 | 2 # from X2 
    3 | 6 # from X1 

我该如何在熊猫中实现这一目标？

Answer 1

使用apply

In [3703]: df.apply(lambda x: x['X%s' % x.Z], axis=1) 
Out[3703]: 
0 5 
1 2 
2 6 
dtype: int64 

In [3706]: df[['idx']].assign(X=df.apply(lambda x: x['X%s' % x.Z], axis=1)) 
Out[3706]: 
    idx X 
0 1 5 
1 2 2 
2 3 6 

Answer 2

注意：按零，当标签不正确的顺序，下面是不行的。

df.set_index('idx').apply(lambda x : x[x['Z']-1],axis=1) 
Out[959]: 
idx 
1 5 
2 2 
3 6 
dtype: int64