基于DataFrame生成字典实现retail_store_id与药品名称的映射

要实现retail_store_id到对应药品名称列表的字典映射,用pandas操作即可快速完成,示例代码如下:
import pandas as pd # 替换为你自己的DataFrame变量名、对应药品名称列名 result_dict = df.groupby("retail_store_id")["药品名称列名"].agg(list).to_dict() # 如果需要门店ID为字符串格式(和参考示例一致),补充以下转换代码 result_dict = {str(key): val for key, val in result_dict.items()}
最终输出的字典结构和你给出的参考完全一致:{'53': ['PAN 40MG TAB', 'MOXIKIND CV 625MG TAB', ......]}
内容的提问来源于stack exchange,提问作者Jahnavi K
相关产品推荐
相关产品推荐

