Numpy：负轴

Question

我想知道“如果轴为负，它从最后一轴到第一轴计数。”在 docs中表示，我测试了这些：

>>> t
array([[1, 2],
       [3, 4]])
>>> np.sum(t, axis=1)
array([3, 7])
>>> np.sum(t, axis=0)
array([4, 6])
>>> np.sum(t, axis=-2)
array([4, 6])

仍然困惑，我需要一些容易理解的解释。

Answer 1

首先查看长度为2的列表上的列表索引：

>>> L = ['one', 'two']
>>> L[-1]  # last element
'two'
>>> L[-2]  # second-to-last element
'one'
>>> L[-3]  # out of bounds - only two elements in this list
# IndexError: list index out of range

axis参数类似，只是它指定了ndarray的维度。使用非正方形数组会更容易看到：

>>> t = np.arange(1,11).reshape(2,5)
>>> t
array([[ 1,  2,  3,  4,  5],
       [ 6,  7,  8,  9, 10]])
>>> t.ndim  # two-dimensional array
2
>>> t.shape  # a tuple of length t.ndim
(2, 5)

那么让我们看一下调用sum的各种方法：

>>> t.sum()  # all elements
55
>>> t.sum(axis=0)  # sum over 0th axis i.e. columns
array([ 7,  9, 11, 13, 15])
>>> t.sum(axis=1)  # sum over 1st axis i.e. rows
array([15, 40])
>>> t.sum(axis=-2)  # sum over -2th axis i.e. columns again (-2 % ndim == 0)
array([ 7,  9, 11, 13, 15])

尝试t.sum(axis=-3)将会出错，因为此数组中只有2个维度。不过，您可以在3d数组上使用它。