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
提供了比标准索引更通用的功能,但它也具有较少的优化,并且可能会显著变慢。