Перевод столбца в строки
Работаю с файлом, загруженным в Jupyter Notebook, использую в основном Pandas.
Есть таблица (данные в файле CSV):
Количество продуктов у одного покупателя не ограничено.
Необходимо получить:
Пробовала и merch, и pivot, но, видимо, моих знаний пока не достаточно.
Ответы (1 шт):
Автор решения: MaxU
→ Ссылка
res = (df
.assign(x=df.groupby("customer").cumcount()+1)
.pivot(index="customer", columns="x", values="product")
.add_prefix("product_")
.rename_axis(None)
.rename_axis(None, axis=1))
результат:
In [16]: res
Out[16]:
product_1 product_2 product_3 product_4 product_5 product_6 product_7
a apple lemon NaN NaN NaN NaN NaN
b kiwi apple NaN NaN NaN NaN NaN
c grapes lemon NaN NaN NaN NaN NaN
d banana apple kiwi lemon melon NaN NaN
e banana kiwi melon NaN NaN NaN NaN
f plum melon pear mango orange banana kiwi