想把X,y中所有y值是最大值或最小值的数据去掉

来源:5-9 使用scikit-learn解决回归问题

Code诗者

2021-02-19

from sklearn.datasets import load_boston
import numpy as np

boston=load_boston()
X=boston.data
y=boston.target
y_max=np.max(y)
y_min=np.min(y)

X=X[all([y<y_max,y>y_min])]
y=y[all([y<y_max,y>y_min])]

print(X.shape,y.shape)

想把X,y中所有y值是最大值或最小值的数据去掉
但报错,按提示的改成all还是不行

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-13-f7e4114156f1> in <module>
      8 y_min=np.min(y)
      9 
---> 10 X=X[all([y<y_max,y>y_min])]
     11 y=y[all([y<y_max,y>y_min])]
     12 

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()


写回答

1回答

liuyubobobo

2021-02-19

这样写:


X = X[(y < y_max) & (y > y_min)]
y = y[(y < y_max) & (y > y_min)]


继续加油!:)

0
0

Python3入门机器学习 经典算法与应用  

Python3+sklearn,兼顾原理、算法底层实现和框架使用。

5893 学习 · 2454 问题

查看课程