Получение данных из столбца на основе индексов в другом столбце в Pandas

Question

У меня есть фрейм данных, в котором два столбца содержат списки, первый содержит элементы, а второй - индексы.Например:

        A             B

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

Я хочу создать еще один столбец, который содержит элементы серии A на основе индекса в B, поэтому результирующий ряд должен выглядеть следующим образом:

     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]