import pandas as pd # 读取数据 data1 = pd.read_csv('D:\运筹优化\杉数科技比赛\data\dc_inventory.csv') # 使用pivot_table函数透视数据,填充缺失值为0 pivoted = data1.pivot_table(values='available_inv', index=['dc_id'], columns=['sku_id'], fill_value=0) # 将透视后的数据转化为列表 results = pivoted.values.tolist() print(results)中每个仓库的产品数量是多少
时间: 2023-07-15 17:12:48 浏览: 402
快速解决pandas.read_csv()乱码的问题
5星 · 资源好评率100%
在你提供的代码中,使用了 pandas 库的 pivot_table 函数将原始数据透视成了以仓库ID(dc_id)为行索引,产品ID(sku_id)为列索引,可用库存数量(available_inv)为值的二维表格。而 results 变量则将透视后的数据转换为了一个列表。
因此,results 列表中每个元素都是一个仓库的产品可用库存数量列表,列表的长度就是产品的数量。你可以通过 results 列表中的每个子列表来获取每个仓库的产品数量。例如,results[0] 就是第一个仓库中每个产品的可用库存数量列表。
阅读全文