python - 按值屏蔽数组，然后屏蔽相应的矩阵

Question

我有一个 numpy 数组 y [1,2,3,4,5,6,1,2,3,4,5,6]

然后我有一个 numpy csr 格式的矩阵 X。

1) 我需要屏蔽 y 中的元素 6。然后，我需要屏蔽 X 中的相应行。

所以，y.shape 是 12。应该是 10。X 是 12,20。应该是 10,20。

如何在 numpy 中执行此操作

Answer 1

如果您的“屏蔽”输出 y 应小于您的输入，则说明您并未真正使用屏蔽。

按照您的 previous question 中的建议，您确实可以轻松找到 y 与 6 不同的索引

condition = (y != 6)

这是一个 bool 数组，您可以使用它来检索不为 6 的 y 值

y = y[condition]

您可以使用相同的条件来获取X的相应行，但它是CSR，因此不支持花哨的索引格式。您仍然可以将其转换为 LIL，然后再转换回来。

您还可以获得 y !=6 的索引

(indices,) = np.nonzero(y != 6)

它是一个常规整数数组，可用于索引 X。