如何将多个列分配给一系列列表熊猫

Question

我有一个熊猫数据框架如下:

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

我想要填充A和b列上从索引0和1到索引2和3的组合列，结果如下:

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

What I tried:

df.loc[df.index.isin([2,3]), ['A','B']] = df.loc[df.index.isin([0,1]),'combined']

当我尝试上面的代码时，输出如下:

A   B      combined
0   1   2        [1,2]
1   2   3        [2,3]
2  NAN  NAN      [3,4]
3  NAN  NAN      [4,5]

请建议。请给我最优的解决方案，因为我正在处理的实际数据以百万计。

提前谢谢你

Answer 1

您可以尝试使用tolist删除索引对齐的数据框架分配

df.loc[df.index.isin([2,3]), ['A','B']] = 
df.loc[df.index.isin([0,1]),'combined'].to_numpy().tolist()