熊猫数据框从另一列中选择列表值

Question

每个人！ 我有一个像这样的熊猫数据框：

        A       B
   0    [1,2,3] 0
   1    [2,3,4] 1

如我们所见，A 列是一个列表，B 列是一个索引值。 我想从 A 得到一个 B 索引的 C 列：

        A       B     C
   0    [1,2,3] 0     1
   1    [2,3,4] 1     3

有没有什么优雅的方法来解决这个问题？ 谢谢！

Answer 1

将列表理解与索引结合使用：

df['C'] = [x[y] for x, y in df[['A','B']].to_numpy()]

或DataFrame.apply ，但如果大型 DataFrame 应该更慢：

df['C'] = df.apply(lambda x: x.A[x.B], axis=1)

---

print (df)
           A  B  C
0  [1, 2, 3]  0  1
1  [2, 3, 4]  1  3