np.ravel_multi_index()用法

arr = np.array([[3,6,6],[4,5,1]])
np.ravel_multi_index(arr, (7,6))
# output: array([22, 41, 37])

[3，6，6]为横座标，[4,5,1]为纵座标索引
 0  1   2   3   4   5   6
0
1
2
3
4
5
如(3,4)位于第4行，第5列，(4-1)*(7-1)+(5-1)=22，即为输出的array中的第一个数字。