如何在Python中减去多维数组？

Question

我有一个多维数组：

n = [[1], [2], [3], [4], [5], [6], [7, 10], [8], [9], [7, 10]]

我希望将它们全部减1。因此结果将是：

result = [[0], [1], [2], [3], [4], [5], [6, 9], [7], [8], [6, 9]]

- colintobing

---

考虑使用NumPy。它在矩阵运算方面比使用列表的列表要好得多。 - user2357112

@user2357112 Numpy通常很好用，但对于这种特殊情况并不适用 - 因为嵌套列表的长度不同，np.array(n)会返回一个dtype=object的数组，而不是int类型，因此np.array(n) - 1会出现TypeError错误（如果它们的大小相同，则会得到正确的答案）。 - lvc

Answer 1

如果你有一个列表嵌套的列表，可以使用嵌套的列表推导式：

In [13]: result = [[e-1 for e in i] for i in n]

In [14]: print result
[[0], [1], [2], [3], [4], [5], [6, 9], [7], [8], [6, 9]]

- zhangxaochen

Answer 2

for x in n:
    for i, y in enumerate(x):
        x[i] = y - 1

至少从空间上来说，这更加高效。

或者如果你想像zhangxaochen一样使用嵌套列表推导式，但将其分配给相同的值，以便它可以原地完成：

n[:] = [[b - 1 for b in a] for a in n]

请注意，这实际上仍然创建了两个额外的列表，因此它具有与分配给新数组相同的空间复杂度。

- Mdev

Answer 3

result=n
for a in range(len(n)):
    for b in range(len(n[a])):
        result[a][b]=n[a][b]-1

- user3471885

---

result=n 只是将其余部分与 n 赋值相同。你可能想要的是 result 应该是 n 的深拷贝，这将需要 result = [m[:] for m in n]。 - lvc

Answer 4

或者您可以使用map()函数：

>>> import operator
>>> n = [[1], [2], [3], [4], [5], [6], [7, 10], [8], [9], [7, 10]]
>>> map(lambda x: map(lambda y: operator.sub(y, 1), x), n)
[[0], [1], [2], [3], [4], [5], [6, 9], [7], [8], [6, 9]]

- alecxe

Answer 5

def difference(a, n):
    try:
        return a - n
    except TypeError:
        return [difference(i, n) for i in a]

>>> difference([[1], [2], [3], [4], [5], [6], [7, 10], [8], [9], [7, 10]], 1)
[[0], [1], [2], [3], [4], [5], [6, 9], [7], [8], [6, 9]]
>>> difference([3, [1, 9, [1, 2, [3, 4, [5, 6]]]], [2], [[3, 4], [5, 6]], [3], [7, 10]], 1)
[2, [0, 8, [0, 1, [2, 3, [4, 5]]]], [1], [[2, 3], [4, 5]], [2], [6, 9]]

它适用于所有多维列表。

- Qrtn