Python numpy.ix_()用法及代码示例

numpy.ix_()函数根据多个序列构造一个开放网格。此函数采用N个一维序列，并返回N个输出，每个输出具有N个维度，因此，除了一个维度外，其他所有形状均为1，并且具有非单位形状值的维度在所有N个维度中循环。

用法： numpy.ix_(args)

参数：

args :[1-D sequences] Each sequence should be of integer or boolean type.

Return :[tuple of ndarrays] N arrays with N dimensions each, with N the number of input sequences. Together these arrays form an open mesh.

代码1：

# Python program explaining 
# numpy.ix_() function 
  
# importing numpy as geek  
import numpy as geek 
  
gfg = geek.ix_([0, 1], [2, 4]) 
  
print (gfg)

输出：

(array([[0],
       [1]]), array([[2, 4]]))

代码2：

# Python program explaining 
# numpy.ix_() function 
  
# importing numpy as geek  
import numpy as geek 
  
arr = geek.arange(10).reshape(2, 5) 
print("Initial array:\n", arr)  
  
ixgrid = geek.ix_([0, 1], [2, 4]) 
  
print("New array:\n", arr[ixgrid])

输出：

Initial array:
 [[0 1 2 3 4]
 [5 6 7 8 9]]
New array:
 [[2 4]
 [7 9]]

相关用法

注：本文由纯净天空筛选整理自sanjoy_62大神的英文原创作品 numpy.ix_() function | Python。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。