Pandas Dataframe 根据列值将值展平到单元格

Question

我有一个从 Excel 读取的简单数据框。为了进一步处理，我需要将“存储”值组合到一个与给定区域对应的单元格中的列表中。为了澄清，我希望每个区域只有一行。在相应的“商店”列中将列出一个单元格中的所有相应商店。

当前状态

期望的状态

我尝试实现了melt，但没有成功。

store_df  = pd.read_excel("Zones_by_Store.xlsx")
store_df.groupby(store_df['Price Zone Name'])
pd.melt(store_df, id_vars=['Price Zone Name'], value_vars=['Store No'])
store_df.to_csv('Stores.csv')

Answer 1

试试这个：

import pandas as pd

price_zone = ['CA2', 'CA2', 'CA2']
store_num = [112, 162, 726]
df = pd.DataFrame(price_zone, columns=['Price Zone'])
df['Store No'] = store_num

df = (df.groupby(['Price Zone']).agg({'Store No': lambda x: x.tolist()}).reset_index())

print(df)