求助解决ValueError：Series真值歧义问题——数据分类函数报错

问题描述 #

编写数据分类函数时触发ValueError，错误提示：
The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all()

代码示例 #

import pandas as pd

df=pd.read_csv(r'C:\Users\gabri\Downloads\credit_scoring_eng.csv')


def economic_class(valor):
    if valor<=16000:
        return 'economic_class'
    elif valor<=24000:
        return 'executive_class'
    elif valor<=32000:
        return 'first_class'
    else:
        return 'five_star_class'
df['class'] = df['total_income'].apply(economic_class)

完整报错信息 #

Cell In [172], line 3
      1 # Criar coluna com categorias
      2 print(df['total_income'])
----> 3 df['class'] = df['total_income'].apply(economic_class)

File c:\Users\gabri\AppData\Local\Programs\Python\Python39\lib\site-packages\pandas\core\series.py:4433, in Series.apply(self, func, convert_dtype, args, **kwargs)
   4323 def apply(
   4324     self,
   4325     func: AggFuncType,
   (...)
   4328     **kwargs,
   4329 ) -> DataFrame | Series:
   4330     """
   4331     Invoke function on values of Series.
   4332 
   (...)
   4431     dtype: float64
   4432     """
-> 4433     return SeriesApply(self, func, convert_dtype, args, kwargs).apply()

File c:\Users\gabri\AppData\Local\Programs\Python\Python39\lib\site-packages\pandas\core\apply.py:1082, in SeriesApply.apply(self)
   1078 if isinstance(self.f, str):
   1079     # if we are a string, try to dispatch
...
   1528         f"The truth value of a {type(self).__name__} is ambiguous. "
   1529         "Use a.empty, a.bool(), a.item(), a.any() or a.all()."
   1530     )

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

需求：尝试过多种网上方案无效，希望实现为DataFrame新增分类列的功能。
（附df.head()截图：）

问题原因 #

报错核心是apply执行时，部分传入economic_class的valor并非单个数值，而是Pandas Series对象。通常因total_income列存在空值（NaN）或数据类型异常，导致函数内的比较操作（valor<=16000）变为Series与标量的比较，触发布尔值歧义错误。

修复方案 #

方案1：使用pd.cut（推荐，原生高效） #

pd.cut是Pandas专为数值区间划分设计的方法，代码简洁且性能更优：

import pandas as pd

df = pd.read_csv(r'C:\Users\gabri\Downloads\credit_scoring_eng.csv')

# 定义区间边界和对应分类标签
bins = [-float('inf'), 16000, 24000, 32000, float('inf')]
labels = ['economic_class', 'executive_class', 'first_class', 'five_star_class']

# 生成分类列，include_lowest确保左区间包含边界值
df['class'] = pd.cut(df['total_income'], bins=bins, labels=labels, include_lowest=True)

方案2：修复原函数并处理异常 #

若坚持使用apply，需在函数中加入类型判断和空值处理：

import pandas as pd

df = pd.read_csv(r'C:\Users\gabri\Downloads\credit_scoring_eng.csv')

def economic_class(valor):
    # 处理空值情况
    if pd.isna(valor):
        return 'unknown'  # 可自定义空值对应的标签
    # 确保仅对数值类型执行比较
    if isinstance(valor, (int, float)):
        if valor <= 16000:
            return 'economic_class'
        elif valor <= 24000:
            return 'executive_class'
        elif valor <= 32000:
            return 'first_class'
        else:
            return 'five_star_class'
    else:
        return 'unknown'

df['class'] = df['total_income'].apply(economic_class)

额外检查建议 #

先排查total_income列的数据类型和空值情况：

# 查看数据类型
print(df['total_income'].dtype)
# 统计空值数量
print(df['total_income'].isna().sum())

若存在非数值类型，先做类型转换：

# 将非数值转换为NaN，保留有效数值
df['total_income'] = pd.to_numeric(df['total_income'], errors='coerce')

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