Python:如何使用列中的列表进行分组

Question

我正在处理以下数据框:

   list          orders
[item1, item2]     A
[item10, item1]    A
[item10, item15]   B

当我对订单进行分组时，列表列完全消失了。有没有一种方法可以使用列表进行分组。我的预期结果是:

      list:                        orders:
[item1, item2, item10, item1]        A
[item10, item15]                     B

Answer 1

使用groupby具有用于展平嵌套列表的自定义 lambda 函数:

from  itertools import chain
f = lambda x: list(chain.from_iterable(x))

或者:

f = lambda x: [z for y in x for z in y]

或者:

f = lambda x: np.concatenate(x.values)

---

df = df.groupby('orders')['list'].apply(f).reset_index()
print (df)
  orders                           list
0      A  [item1, item2, item10, item1]
1      B               [item10, item15]

如果性能不重要(大数据慢):

df = df.groupby('orders')['list'].sum().reset_index()