计算3个月采购平均值并判定罚款时遇ValueError问题求助

场景与问题 #

需求：若Excel中人员3个月的平均采购额低于60，则处以12万美元罚款。编写的代码如下：

import pandas as pd

file_df = pd.read_excel('users.xlsx')

def penalty_calculation(df):
    sum_of_three_month = df["First month purchase "] + df["Purchase of the second month"] + df["Purchase of the third month"]
    df["Average purchase of 3 months"] = sum_of_three_month // 3
    if df["Average purchase of 3 months"] <= 60:
        df["penalty"] = "$120000"

penalty_calculation(file_df)

运行时抛出错误：

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

错误原因 #

核心问题在于你在if语句中直接对整个Pandas Series做布尔判断：

• df["Average purchase of 3 months"] <= 60会返回一个由布尔值组成的Series，每个元素对应原数据中一行的判断结果（比如某行平均是50则为True，某行是70则为False）。
• Python的if语句只能接受单个布尔值，无法直接处理包含多个布尔值的Series，因此会抛出"真值不明确"的错误。

修正方法 #

你需要用Pandas的向量化操作来处理整列数据，推荐两种方式：

方式1：使用loc定位符合条件的行 #

import pandas as pd

file_df = pd.read_excel('users.xlsx')

def penalty_calculation(df):
    # 计算三个月总和与浮点平均值（替换//为/，避免整数除法导致的计算偏差）
    sum_of_three_month = df["First month purchase "] + df["Purchase of the second month"] + df["Purchase of the third month"]
    df["Average purchase of 3 months"] = sum_of_three_month / 3
    # 给符合条件的行设置罚款金额
    df.loc[df["Average purchase of 3 months"] <= 60, "penalty"] = "$120000"
    # 给不符合条件的行填充默认值（可选）
    df["penalty"] = df["penalty"].fillna("")

penalty_calculation(file_df)

方式2：使用numpy.where实现条件赋值 #

import pandas as pd
import numpy as np

file_df = pd.read_excel('users.xlsx')

def penalty_calculation(df):
    sum_of_three_month = df["First month purchase "] + df["Purchase of the second month"] + df["Purchase of the third month"]
    df["Average purchase of 3 months"] = sum_of_three_month / 3
    # 一行代码完成条件赋值：符合条件则设为罚款金额，否则为空字符串
    df["penalty"] = np.where(df["Average purchase of 3 months"] <= 60, "$120000", "")

penalty_calculation(file_df)

额外提示 #

原代码中使用//（整数除法）计算平均值，可能会导致结果偏差（比如总和179时，179//3=59，但实际平均是59.67），建议替换为/得到浮点型平均值，更贴合业务逻辑。

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