python - SQL 输出到列表

Question

我有一个这种格式的 csv

ORDER_ID    Item_DESC
 121    Beer
 121    Chips
 121    Wine
 141    Chips
 141    Wine

我需要以这种格式推出它:

121    Beer,Chips,Wine
141    Chips,Wine

有人能给我指出正确的方向吗，我已经伤透了一周了。

我尝试了以下方法:

Excel Pandas-枢轴、堆叠、熔化 SQL 中的递归表

感谢任何帮助。

Answer 1

您可以使用groupby and apply :

In [11]: df
Out[11]:
   ORDER_ID Item_DESC
0       121      Beer
1       121     Chips
2       121      Wine
3       141     Chips
4       141      Wine

In [12]: df.groupby("ORDER_ID")["Item_DESC"].apply(",".join)
Out[12]:
ORDER_ID
121         Beer,Chips,Wine
141              Chips,Wine
Name: Item_DESC, dtype: object

然后使用 to_csv就这个结果而言。