numpy.ix_

numpy.ix_(*args)

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

此函数接受n个一维序列，并返回n个输出，每个输出都有n个维度，这样除了一个维度外，形状都是1，而具有非单位形状值的维度在所有n个维度中循环。

使用 ix_ 我们可以快速构造索引数组来索引交叉积。 a[np.ix_([1,3],[2,5])] 返回数组 [[a[1,2] a[1,5]], [a[3,2] a[3,5]]] .

参数

args一维序列

每个序列都应该是整数或布尔类型。布尔序列将被解释为对应维度的布尔掩码（相当于传入 np.nonzero(boolean_sequence) ）

返回

outN数组的元组

n个数组，每个数组有n个维度，输入序列的数目为n。这些数组一起形成一个开放的网格。

参见

ogrid, mgrid, meshgrid

实例

>>> 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]])

>>> ixgrid = np.ix_([True, True], [2, 4])
>>> a[ixgrid]
array([[2, 4],
       [7, 9]])
>>> ixgrid = np.ix_([True, True], [False, False, True, False, True])
>>> a[ixgrid]
array([[2, 4],
       [7, 9]])

numpy.indices numpy.ravel_multi_index