np.ravel_multi_index()用法
Posted zyl-hub
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了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中的第一个数字。
以上是关于np.ravel_multi_index()用法的主要内容，如果未能解决你的问题，请参考以下文章
解释内存中的栈(stack)堆(heap)和方法区(method area) 的用法？