当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python numpy ix_用法及代码示例


本文简要介绍 python 语言中 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) )。

返回

out ndarray 的元组

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

例子

>>> 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.org大神的英文原创作品 numpy.ix_。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。