numpy.ix_#
- numpy.ix_(*args)[source]#
从多个序列构建一个开放网格。
此函数接收 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)
)。
- 返回值:
- outndarray 元组
N 个数组,每个数组具有 N 个维度,其中 N 是输入序列的数量。这些数组一起形成一个开放网格。
示例
>>> import numpy as np >>> 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]])