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中的第一個數字。