从矩阵中删除某些元素

Question

我有以下问题：

我有一个矩阵。现在，我想在矩阵的每一行中删除一个条目：在包含某个数字（比如4）的行中，我想删除带有该数字的条目，在其他行中，我只想删除最后一个元素。

如果我有矩阵

    matrix=np.zeros((2,2))
    matrix[0,0]=2
    matrix[1,0]=4
    matrix

这给了

2 0

4 0

删除后应该只是

2.

0

谢谢你的帮助！

Answer 1

所以，假设一行中最多只有一个4，您要做的是：

1. 迭代所有行，如果有四个使用roll，那么它将成为最后一个元素
2. 删除最后一列

在具有4的行中，它将删除这个4并移动它之后的剩余值，在没有4的行中，它将删除最后一个元素。

（我冒昧地尝试了一个更大的矩阵，以确保输出符合预期）

试试这个：

import numpy as np


# Actual solution
def remove_in_rows(mat, num):
    for i, row in enumerate(mat):
        if num in row.tolist():
            index = row.tolist().index(num)
            mat[i][index:] = np.roll(row[index:], -1)
    return np.delete(mat, -1, 1)


# Just some example to demonstrate it works
matrix = np.array([[10 * y + x for x in range(6)] for y in range(6)])
matrix[1, 2] = 4
matrix[3, 3] = 4
matrix[4, 0] = 4

print("BEFORE:")
print(matrix)

matrix = remove_in_rows(matrix, 4)

print("AFTER:")
print(matrix)

输出：

BEFORE:
[[ 0  1  2  3  4  5]
 [10 11  4 13 14 15]
 [20 21 22 23 24 25]
 [30 31 32  4 34 35]
 [ 4 41 42 43 44 45]
 [50 51 52 53 54 55]]
AFTER:
[[ 0  1  2  3  5]
 [10 11 13 14 15]
 [20 21 22 23 24]
 [30 31 32 34 35]
 [41 42 43 44 45]
 [50 51 52 53 54]]