Python numpy.indices()函数

numpy.indices()函数返回一个代表网格索引的数组。计算一个数组，其中子数组包含的索引值0，1，…只沿相应的轴线变化。

语法: numpy.indices(dimensions, dtype, sparse = False)

参数 :
dimensions : [ints序列] 网格的形状。
dtype: [dtype, optional] 结果的数据类型。
sparse: [boolean, optional] 返回网格的稀疏表示而不是密集表示。默认为假。

返回 : [ndarray 或 ndarrays 的元组]
如果sparse为False。
返回一个网格索引的数组，grid.shape = (len(dimensions), )+ tuple(dimensions)。

如果sparse为True。
返回一个数组的元组，grid[i].shape = (1, …, 1, dimensions[i], 1, …, 1)，dimensions[i]在第i个位置。

代码#1：

# Python program explaining
# numpy.indices() function
       
# importing numpy as geek 
import numpy as geek 
   
gfg = geek.indices((2, 3))
  
print (gfg)

输出 :

[[[0 0 0]
  [1 1 1]]

 [[0 1 2]
  [0 1 2]]]

代码#2：

# Python program explaining
# numpy.indices() function
       
# importing numpy as geek 
import numpy as geek 
   
grid = geek.indices((2, 3))
gfg = grid[1]
  
print (gfg)

输出 :

[[0 1 2]
 [0 1 2]]