首页 > 解决方案 > 如何过滤给定特定字母的ndarray?

问题描述

我有一个ndarray (y)包括现有标签。这些标签是二进制 0 或 1。我有第二个ndarray (X)表示训练集。

如何从标签0对应的所有数据中提取。所以,新数据将是:原始数据:X和 另一个提取与其对应的数据。yX, yX1y1

标签: numpynumpy-ndarray

解决方案


您可以使用:

X1, y1 = X[y==0], y[y==0]

推荐阅读