python - 为 Pandas 中的每个键创建一个值列表？

Question

我有一个 CSV 文件，看起来像这样，我已将其加载到数据框中，

keycode, warehouse_id
      1,          WH1
      1,          WH2
      1,          WH1

我想将其映射到看起来像这样的输出

keycode, warehouse_ids
      1,    [WH1, WH2]

我不确定在 pandas 中从哪里开始？我尝试使用数据透视表但使用聚合但无法选择正确的聚合函数。

提前致谢。

Answer 1

使用groupby + unique :

df1 = df.groupby('keycode,')['warehouse_id'].unique().reset_index()

print(df1)
  keycode, warehouse_id
0       1,   [WH1, WH2]

---

解释:

通常，在使用 groupby 和如下所示的单个操作列时会产生 Series索引为 groupby key，为此我们使用 reset_index将 index 更改为 column:

print(df.groupby('keycode,')['warehouse_id'].unique())
keycode,
1,    [WH1, WH2]
Name: warehouse_id, dtype: object

print(type(df.groupby('keycode,')['warehouse_id'].unique()))
<class 'pandas.core.series.Series'> 

print(df.groupby('keycode,')['warehouse_id'].unique().reset_index())
  keycode, warehouse_id
0       1,   [WH1, WH2]