python - 如何在 numpy 中迭代时删除行

Question

如何像 Java 一样在 numpy 中迭代时删除行:

Iterator < Message > itMsg = messages.iterator();
while (itMsg.hasNext()) {
    Message m = itMsg.next();
    if (m != null) {
        itMsg.remove();
        continue;
    }
}

这是我的伪代码。迭代时删除条目全部为 0 和 1 的行。

#! /usr/bin/env python
import numpy as np

M = np.array(
    [
        [0, 1 ,0 ,0],
        [0, 0, 1, 0],
        [0, 0, 0, 0], #remove this row whose entries are all 0
        [1, 1, 1, 1]  #remove this row whose entries are all 1
    ])


it = np.nditer(M, order="K", op_flags=['readwrite'])
while not it.finished :
    row = it.next()     #how to get a row?
    sumRow = np.sum(row)
    if sumRow==4 or sumRow==0 : #remove rows whose entries are all 0 and 1 as well
        #M = np.delete(M, row, axis =0)
        it.remove_axis(i)  #how to get i?

Answer 1

编写好的 numpy 代码需要您以矢量化的方式思考。并非每个问题都具有良好的矢量化，但对于那些具有良好矢量化的问题，您可以非常轻松地编写干净且快速的代码。在这种情况下，我们可以决定要删除/保留哪些行，然后使用它来索引您的数组:

>>> M
array([[0, 1, 0, 0],
       [0, 0, 1, 0],
       [0, 0, 0, 0],
       [1, 1, 1, 1]])
>>> M[~((M == 0).all(1) | (M == 1).all(1))]
array([[0, 1, 0, 0],
       [0, 0, 1, 0]])

<小时/>

一步一步，我们可以将 M 与某些东西进行比较以生成 bool 数组:

>>> M == 0
array([[ True, False,  True,  True],
       [ True,  True, False,  True],
       [ True,  True,  True,  True],
       [False, False, False, False]], dtype=bool)

我们可以使用all来查看一行或一列是否全部为真:

>>> (M == 0).all(1)
array([False, False,  True, False], dtype=bool)

我们可以使用|进行或操作:

>>> (M == 0).all(1) | (M == 1).all(1)
array([False, False,  True,  True], dtype=bool)

我们可以用它来选择行:

>>> M[(M == 0).all(1) | (M == 1).all(1)]
array([[0, 0, 0, 0],
       [1, 1, 1, 1]])

但是由于这些是我们想要丢弃的行，因此我们可以使用 ~ (NOT)来翻转 False 和 True:

>>> M[~((M == 0).all(1) | (M == 1).all(1))]
array([[0, 1, 0, 0],
       [0, 0, 1, 0]])

<小时/>

如果我们想保留不全部为 1 或全部 0 的列，我们只需更改我们的轴即可正在研究:

>>> M
array([[1, 1, 0, 1],
       [1, 0, 1, 1],
       [1, 0, 0, 1],
       [1, 1, 1, 1]])
>>> M[:, ~((M == 0).all(axis=0) | (M == 1).all(axis=0))]
array([[1, 0],
       [0, 1],
       [0, 0],
       [1, 1]])