Python循环分配周次不符合容量限制问题求助

Hey there! Let's dig into why your weekly assignment loop is cutting short and not respecting those capacity limits. I've dealt with similar batch allocation problems before, so let's break this down step by step.

常见问题根源推测 #

从你的描述来看，循环提前终止、分配不符合限制，大概率是以下几个逻辑点出了问题：

• 循环终止条件错误：比如你可能只判断了"当前周容量是否已满"，而没确认"所有行是否都完成分配"，导致还有未处理的行时循环就停了
• 容量计数逻辑漏洞：没有在每周开始时重置等级计数器，或者计数时漏掉了部分未分配行
• 分配状态判断失误：错误标记了行的分配状态，导致后续循环漏处理未分配数据

针对性排查步骤 #

1. 检查循环终止条件 #

你的循环应该持续到所有行都分配了周次，而不是"当前周的容量用满就停"。比如如果原代码是类似while current_week_capacity < max_total，那肯定会提前终止——正确的条件应该是判断是否还有未分配的行。

2. 重置每周的容量计数器 #

每次进入新周时，必须把HIGH/MED/LOW的计数清零，否则会累计上周的数量，导致刚进入新周就触发容量上限，直接跳周。

3. 验证行的分配状态标记 #

确保你给已分配周次的行做了明确标记（比如给week字段赋值），并且后续循环能准确过滤出未标记的行，避免漏处理。

修正后的示例代码（基于Pandas） #

假设你的CSV数据有level字段存储等级，我们用Pandas实现符合要求的分配逻辑：

import pandas as pd
import math

# 加载数据集
df = pd.read_csv("your_dataset.csv")
# 初始化周次列，未分配标记为NaN
df["assigned_week"] = pd.NA

# 预设容量限制
level_capacity = {"HIGH": 30, "MED": 100, "LOW": 100}
current_week = 1

# 循环直到所有行完成分配
while df["assigned_week"].isna().any():
    # 每周开始时重置各等级计数
    high_count = 0
    med_count = 0
    low_count = 0
    
    # 遍历所有未分配的行
    for idx, row in df[df["assigned_week"].isna()].iterrows():
        row_level = row["level"]
        
        # 根据等级容量判断是否分配当前周
        if row_level == "HIGH" and high_count < level_capacity["HIGH"]:
            df.loc[idx, "assigned_week"] = current_week
            high_count += 1
        elif row_level == "MED" and med_count < level_capacity["MED"]:
            df.loc[idx, "assigned_week"] = current_week
            med_count += 1
        elif row_level == "LOW" and low_count < level_capacity["LOW"]:
            df.loc[idx, "assigned_week"] = current_week
            low_count += 1
        # 容量满的等级跳过，留到下一周分配
    
    # 当前周分配完毕，进入下一周
    current_week += 1

# 验证分配结果
print(f"总分配行数：{df['assigned_week'].notna().sum()} (原数据集总行数：{len(df)})")
print("\n每周各等级分配统计：")
weekly_stats = df.groupby(["assigned_week", "level"]).size().unstack(fill_value=0)
print(weekly_stats)

# 检查是否有超出容量的情况
print("\n容量合规性检查：")
print(weekly_stats["HIGH"].le(30).all(), "HIGH等级所有周都符合≤30的限制")
print(weekly_stats["MED"].le(100).all(), "MED等级所有周都符合≤100的限制")
print(weekly_stats["LOW"].le(100).all(), "LOW等级所有周都符合≤100的限制")

关键逻辑说明 #

• 循环条件df["assigned_week"].isna().any()确保只要还有未分配行就继续执行
• 每周重置计数器，保证每周的容量统计是独立的
• 遍历未分配行时，只在对应等级还有剩余容量时分配当前周，否则留到下一周
• 最后输出的统计和合规性检查，可以快速验证结果是否符合要求

内容的提问来源于stack exchange，提问作者Havard Kleven