SMOTE重采样后调用y.value_counts()报numpy.ndarray无该属性如何解决
报错原因
value_counts()是Pandas独有的方法,仅支持Series和DataFrame类型的对象调用。你在调用oversample.fit_resample(X,y)执行过采样时,imblearn库的接口默认会将输入的特征、标签数据统一转换为Numpy数组格式输出,因此处理后得到的y已经不再是最初的Pandas Series类型,而是numpy.ndarray类型,自然没有value_counts属性,触发报错。
解决方法
你可以选择以下任意一种方案解决问题:
- 方案一:将处理后的标签转回Pandas Series类型后再调用方法
import pandas as pd pd.Series(y).value_counts()
- 方案二:直接使用Numpy自带的统计方法实现相同的计数效果,无需转换类型
import numpy as np np.unique(y, return_counts=True)
返回结果的第一个数组为所有类别值,第二个数组为对应类别的样本数量。
- 方案三:如果需要保留原始列名,可转成DataFrame后统计
pd.DataFrame(y, columns=['quality']).value_counts()
内容的提问来源于stack exchange,提问作者Inosuke_Hashibira
相关产品推荐
相关产品推荐

