Python numpy.ix_()函数
numpy.ix_()函数从多个序列构建一个开放的网格。这个函数接收N个一维序列,并返回N个输出,每个输出有N个维度,这样,除了一个维度外,其他维度的形状都是1,并且具有非单位形状值的维度在所有N个维度中循环。
语法: numpy.ix_(args)
参数 :
args : [1-D序列] 每个序列应该是整数或布尔类型的。
返回: [nndarrays的元组] N个数组,每个数组有N个维度,N为输入序列的数量。这些数组共同构成一个开放的网状结构。
代码#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]]