python - 如何删除对应于 Python X 列中已删除异常值的相应 Y 值

Question

在建模之前，我使用以下方法删除了 X 变量中的异常值：

z = np.abs(stats.zscore(X))
X = X[(z < 3).all(axis=1)]

我怎样才能使 Y 列中的相应值被删除，以便我可以继续我的建模？

标签： pythonmachine-learningdata-cleaningoutliers

Answer 1

您需要保存遮罩，然后将其应用到X和上Y。

z = np.abs(stats.zscore(X))
mask = (z < 3).all(axis=1)
X = X[mask]
Y = Y[mask]