上一主题

numpy.indices

下一主题

numpy.ravel_multi_index

numpy.ix

numpy.ix_(*args)[source]

从多个序列构造一个打开的网格。

该函数采用N 1-D序列并返回N个输出,每个具有N维,使得除了一个维度之外的所有形状均为1,并且具有非单位形状值的维度遍历所有N维。

使用ix_可以快速构建将索引交叉乘积的索引数组。a[np.ix_([1,3],[2,5])] returns the array [[a[1,2] a[1,5]], [a[3,2] a[3,5]]].

参数:

args:1-D序列

返回:

out:ndarrays的元组

N数组,每个具有N维,其中N是输入序列的数目。这些数组一起形成一个开放的网格。

也可以看看

ogridmgridmeshgrid

例子

>>> a = np.arange(10).reshape(2, 5)
>>> a
array([[0, 1, 2, 3, 4],
       [5, 6, 7, 8, 9]])
>>> ixgrid = np.ix_([0,1], [2,4])
>>> ixgrid
(array([[0],
       [1]]), array([[2, 4]]))
>>> ixgrid[0].shape, ixgrid[1].shape
((2, 1), (1, 2))
>>> a[ixgrid]
array([[2, 4],
       [7, 9]])