#产品销售量 data5 = pd.read_csv('D:\运筹优化\杉数科技比赛\data\test_sales_data.csv') # 构建长度为77的数组并初始化为0 PS = [0] * 77 # 根据第一列汇总求和销售量，并将结果填入数组 for i, row in data5.iterrows(): if row['sku_id'][:3] == 'SKU': idx = int(row['sku_id'][3:]) - 1 PS[idx] += row['sale'] # 输出结果 print(PS)在代码中加入以dc_id区分销售量

你可以尝试修改代码如下：

data5 = pd.read_csv('D:\运筹优化\杉数科技比赛\data\test_sales_data.csv')

# 构建长度为77的数组并初始化为0
PS = [0] * 77

# 根据第一列汇总求和销售量，并将结果填入数组
for i, row in data5.iterrows():
    if row['sku_id'][:3] == 'SKU':
        idx = int(row['sku_id'][3:]) - 1
        dc_id = row['dc_id'] - 1 # 获取dc_id并减一得到下标
        PS[idx][dc_id] += row['sale'] # 在对应下标的dc_id上累加销售量

# 输出结果
print(PS)

这样修改后，你就可以根据dc_id分别统计每个仓库的销售量了。其中PS数组的第一维代表商品编号，第二维代表dc_id。