dask.array.Array.vindex

dask.array.Array.vindex

property Array.vindex

使用广播的向量化索引。

这相当于使用相互广播的数组进行 numpy 的高级索引。这允许逐点索引:

>>> import dask.array as da
>>> x = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
>>> x = da.from_array(x, chunks=2)
>>> x.vindex[[0, 1, 2], [0, 1, 2]].compute()
array([1, 5, 9])

混合基本/高级索引与切片/数组也是支持的。结果中的维度顺序遵循为 ndarray.vindex 提出的顺序:由数组跨越的子空间后跟所有切片。

注意:vindex 提供了比标准索引更通用的功能,但它也具有较少的优化,并且可能会显著变慢。