Question

我有一个整数列表，表示矩阵中的位置（centre）。例如，

centre = [2, 50, 100, 89]

我还有两个numpy矩阵，X和Y。如果数字在centre中，我需要删除矩阵中的所有行。我可以这样做：

for each in centre:
  x = numpy.delete(x, (each), axis=0)

然而，由于指数全部都将被淘汰，数字将全部消失。那么，我该怎么做呢？

Answer 1

只需在一个电话中执行delete：

In [266]: B
Out[266]: 
array([[ 2,  4,  6],
       [ 8, 10, 12],
       [14, 16, 18],
       [20, 22, 24]])
In [267]: B1=np.delete(B,[1,3],axis=0)
In [268]: B1
Out[268]: 
array([[ 2,  4,  6],
       [14, 16, 18]])

你的问题有点令人困惑。我假设你想按索引号删除行，而不是某种内容（不像列表find）。

但是，如果你必须迭代（与列表一样）以相反的顺序进行 - 这样索引就不会搞砸了。您可能必须先对索引进行排序（np.delete并不要求）。

In [269]: B1=B.copy()
In [270]: for i in [1,3][::-1]:
     ...:     B1=np.delete(B1,i,axis=0)

必须迭代的列表示例：

In [276]: B1=list(range(10))
In [277]: for i in [1,3,5,7][::-1]:
     ...:     del B1[i]
In [278]: B1
Out[278]: [0, 2, 4, 6, 8, 9]

=============

使用这样的列表输入，np.delete相当于：

In [285]: mask=np.ones((4,),bool)
In [286]: mask[[1,3]]=False
In [287]: mask
Out[287]: array([ True, False,  True, False], dtype=bool)
In [288]: B[mask,:]
Out[288]: 
array([[ 2,  4,  6],
       [14, 16, 18]])

从numpy矩阵

1 个答案: