numpy中多维数组的绝对索引

Posted TQCAI

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了numpy中多维数组的绝对索引相关的知识,希望对你有一定的参考价值。

这涉及到吧多维数组映射为一维数组。

对于3维数组,有公式:

def MAP(x,y,z):
    return y_s * z_s * x + z_s * y + z

此公式可以推广到N维

测试代码:(两个输出相同,说明测试成功)

import numpy as np

x_s=4
y_s=3
z_s=9

def MAP(x,y,z):
    return y_s * z_s * x + z_s * y + z

org=np.arange(x_s*y_s*z_s)
arr=np.resize(org,[x_s,y_s,z_s])

x=2
y=1
z=3
print(org[MAP(x,y,z)])
print(arr[x,y,z])

 

以上是关于numpy中多维数组的绝对索引的主要内容,如果未能解决你的问题,请参考以下文章

numpy多维索引:使用np数组并列出不同的结果

获取多维 NumPy 数组中最大值的位置

NumPy 基础

NumPy 基础

NumPy 基础

在多维数组上使用 numpy.argmax()