带有 2D 切片的 numpy 视图

Question

Numpy 使用视图对象来最小化内存复制。 下面的代码使用索引列表对原始 ndarray 进行切片。 结果是None ，这意味着arr[ [0, 3] ]分配了自己的内存。

arr = np.arange(5)
print(arr[ [0, 3] ].base)

在范围切片的情况下， arr[0:2]返回指向原始 ndarray ( [0 1 2 3 4] ) 的视图

print(arr[ 0:2 ].base)

在 2D 切片的情况下，它的工作方式与我预期的None不同。

arr = np.array([
    [0, 1, 2],
    [3, 4, 5],
    [6, 7, 8],
])

print(arr[  : , [0, 2] ].base )

[[0 3 6]
 [2 5 8]]

我想知道为什么None没有返回，为什么 base 的形状不是 (3, 2)。

Answer 1

In [205]: arr = np.array([
     ...:     [0, 1, 2],
     ...:     [3, 4, 5],
     ...:     [6, 7, 8],
     ...: ])
     ...: 
     ...: arr[  : , [0, 2] ]
Out[205]: 
array([[0, 2],
       [3, 5],
       [6, 8]])

数组索引符合我们的预期。 只是base不同。

In [206]: _.base
Out[206]: 
array([[0, 3, 6],
       [2, 5, 8]])

我认为这个base为索引的基本过程提供了线索。 它使用[0,2]高级索引制作了一个副本，并执行某种转置以返回所需的数组。

我还没有完全注意base 。 如果对某些东西是否是视图有疑问，我喜欢比较__array_interface__ 。

通常知道data值明显不同（而不仅仅是偏移量）就足够了。

In [209]: arr.__array_interface__
Out[209]: 
{'data': (38026368, False),
 'strides': None,
 'descr': [('', '<i8')],
 'typestr': '<i8',
 'shape': (3, 3),
 'version': 3}
In [210]: arr[  : , [0, 2] ].__array_interface__
Out[210]: 
{'data': (38386288, False),
 'strides': (8, 24),
 'descr': [('', '<i8')],
 'typestr': '<i8',
 'shape': (3, 2),
 'version': 3}

进一步看strides ：

In [218]: arr.strides
Out[218]: (24, 8)
In [219]: arr[:,[0,2]].strides
Out[219]: (8, 24)
In [220]: arr[:,[0,2]].copy().strides
Out[220]: (16, 8)

arr strides 是(3*8, 8) ，因为步进到下一行意味着跳过 3 列值。 但是索引的步幅是相反的，这是我对其base的转置所期望的。 完整副本是(2*8,8)反映 2 列形状。

所以这里base揭示了索引过程的细节。 我以前没有注意到这些，而且我已经使用numpy有一段时间了。

---

base反映构建过程的另一种情况是通过arange制作你的arr ：

In [213]: np.arange(9).reshape(3,3).base
Out[213]: array([0, 1, 2, 3, 4, 5, 6, 7, 8])

该base来自arange ，但我们从未分配给变量。