python - 根据 Pandas 中另一列的索引从一列获取数据

Question

我有一个数据框，其中两列包含列表，第一列包含项目，第二列包含索引。喜欢:

        A             B

0   [11, 12, 13, 14]    [0, 2, 3, 0]
1   [11, 22]            [1, 0]
2   [9, 7]              [0, 1]

我想根据 B 中的索引创建另一个包含系列 A 元素的列，因此生成的系列应如下所示:

     C
0  [11,13,14,11]
1  [22,11]
2  [9,7]

有人可以帮忙吗？非常感谢提前

Answer 1

使用

[ np.array(x)[y].tolist() for x , y in  zip(df.A,df.B)]

Out[298]: [[11, 13, 14, 11], [22, 11]]

<小时/>

分配后

df['c']=[ np.array(x)[y].tolist() for x , y in  zip(df.A,df.B)]
df
Out[300]: 
                  A             B                 c
0  [11, 12, 13, 14]  [0, 2, 3, 0]  [11, 13, 14, 11]
1          [11, 22]        [1, 0]          [22, 11]