将具有非唯一索引的 Pandas Dataframe 与一维 Numpy 数组合并

Question

我有一个带有索引的 Pandas 数据框，它不是唯一的。

index   col1  col2 
  1      a     b
  1      r     t
  2      e     e

还有一个与 Dataframe 长度相同的一维 Numpy 数组

数组 = [g, h, y]

我需要将数组添加到新列中，例如：

index   col1  col2  array
  1      a     b     g
  1      r     t     h
  2      e     e     y

我尝试将数组制作成 Pandas 系列和数据框，但由于数组具有不同的索引而遇到麻烦。

我需要在第一个数据帧中保留非唯一索引。我该如何管理？

Answer 1

您可以将数组分配为列：

a = ['g', 'h', 'y']
df['array']=a
print(df)

   index col1 col2 array
0      1    a    b     g
1      1    r    t     h
2      2    e    e     y