numpy.unravel_index 函数功能说明

函数名:numpy.unravel_index(indices, dims)

功能:返回indices中的元素在维度为dims的数组中的索引值,默认按元组的形式返回。

举例:

1)numpy.unravel_index(indices=[22, 41, 37], dims=(7,6)),其返回元组(array([3, 6, 6]), array([4, 5, 1]))。

其过程类似:首先构造一个数组arr1=np.array(range(7*6)).reshape((7,6));返回indices中的元素值在数组arr1中对应值的索引位置。[3, 6, 6]分别对应[22, 41, 37]在第1维中的位置,[4, 5, 1]分别对应[22, 41, 37]在2维中的位置。

2)numpy.unravel_index(indices=1621, dims=(6,7,8,9)),其返回元组(3, 1, 4, 1)。

其过程类似:首先构造一个数组arr2=np.array(range(6*7*8*9)).reshape((6,7,8,9));返回indices中的元素值在数组arr2中对应值的索引位置。(3, 1, 4, 1)分别对应1621在第1维、第2维、第3维和第4维中的位置。

说明:函数numpy.unravel_index(indices, dims)返回的索引值从0开始计数。



 
 

你可能感兴趣的:(numpy)