根据索引从矩阵中删除一些元素

Question

我有一个矩阵：

a = ([[1, 0, 0, 0],
       [0, 0, 1, 1],
       [0, 1, 0, 1]
       [1, 0, 0, 1]])

我想打印矩阵中的 0，但不是所有的 0。我只想在具有最小索引的每一行中保留 0，并删除该行中所有后续的零。 例如，在该矩阵的第一行中，应保留第二个元素 (a[0][1])，而应删除第一行中的其余元素，因为它们全为零。

我将pop() 用于二维数组，但出现属性错误。而且输出也不正确。我不知道如何比较索引并选择每一行中最小的列索引。

这是我的代码：

for ix, row in enumerate(a):
  for iy, i in enumerate(row):
    if i==0 and (iy+ix<(iy+1)+ix) :
        a[ix].pop((iy+1))
        print(ix,iy)
    elif i==0 and (iy+ix>(iy+1)+ix):
        a[ix].pop(iy)
        print(ix,iy+1)
print(a)

我的预期结果是一组索引和修改后的矩阵 a。

0 1

1 0

2 0

3 1

a=[[1,0],[0,1,1],[0,1,1],[1,0]]

谁能帮帮我？

Answer 1

此解决方案仅在每行至少有一个零时才有效。

indices = []
for x,row in enumerate(a):
    i = row.index(0)
    indices.append((x,i))
    a[x] = row[:i+1] + [e for e in row[i:] if e]

print(indices)
print(a)

输出

[(0, 1), (1, 0), (2, 0), (3, 1)]
[[1, 0], [0, 1, 1], [0, 1, 1], [1, 0, 1]]

Answer 2

假设每一行都有一个零，你可以得到它的列索引

c = np.argmin(a, axis=1)

或者，如果矩阵可以包含负数，你可以这样做

c = np.argmax(np.equal(a, 0), axis=1)

行只是

r = np.arange(len(a))

你想要的结果是那么

result = np.stack((r, c), axis=-1)

如果其中有没有零的行，您可以使用掩码过滤结果：

mask = np.array(a)[r, c] == 0
result = result[mask, :]

Answer 3

查看您的示例输入

a = [[1,0,0,0],[0,0,1,1],[0,1,0,1],[1,0,0,1]]

和预期的输出 >>[(0, 1), (1, 0), (2, 0), (3, 1)]

您可以将问题重新定义为在每一行中找到值为零的元素的索引（如果存在多个元素，则返回第一个）。

通过这种方式，解决方案很简单，只需遍历a 的每一行并检索值0 的index（默认情况下只会返回第一个元素）。

使用如下所示的列表推导：

value_to_find = 0
desired_indexes = [
  row.index(value_to_find) for row in a
]

或使用map ：

value_to_find = 0
desired_indexes = map(lambda row:row.index(value_to_find), a)

然后您可以枚举它们以将结果与行号配对

enumerate(desired_indexes)

瞧！ >>[(0, 1), (1, 0), (2, 0), (3, 1)]

整个解决方案可以像这样写在一行中：

answer = list(enumerate(map(lambda row:row.index(0), a)))

Answer 4

试试这个：

a = [[1, 0, 0, 0],
       [0, 0, 1, 1],
       [0, 1, 0, 1],
       [1, 0, 0, 1]]

b = []
for i in a:
    f = False
    c = []
    for j in i:
        if (j==0 and f==False) or j != 0:
            c.append(j)
            if j == 0: f = True
        else:
            continue
    b.append(c)

输出：

[[1, 0], [0, 1, 1], [0, 1, 1], [1, 0, 1]]

要在数组中获取索引zero，您可以试试这个：

list({i : j.index(0) for i,j in enumerate(b)}.items())
# [(0, 1), (1, 0), (2, 0), (3, 1)]