有没有办法将 multi_index 转换为 int 字符串/列表/数组？

Question

我想将 numpy 中给定数组的值更改为数组其他元素的乘积。因此，我想提取 multi_index 并对其进行操作，以便我可以识别位置并使用它。（例如 nditer 遍历所有元素并始终执行 'current position in array = next position +position above in array'

我试图用当前位置的 multi_index 调用一个函数，并希望所述函数接受它，例如增加一个位置。 (<0 , 1> ---> <0 , 2> 而 <0 , n> n>=length 否则 <0 , 1> ---> <1 , 0>)

import numpy as np;

def fill(multi_index):
    "This will get the new value of the current iteration value judgeing from its index"
    return a[(multi_index + <0,1>) + (multi_index + <0,-1>) + (multi_index + <1,0>) + (multi_index + <-1,0>)]

#a = np.random.uniform(0, 100, size=(100, 100))
a = np.arange(6).reshape(2,3)

it = np.nditer(a, flags=['multi_index'], op_flags=['readwrite'])
while not it.finished:
    it[0] = fill(it.multi_index)
    print(it[0])
    it.iternext()

"""for x in np.nditer(a, flags=['multi_index'], op_flags=['readwrite']):

    print(x)"""

我不明白如何从 multi_index 中提取实际的 "coordinates"。我对 python 有点陌生，所以如果可能的话请尽量解释清楚。谢谢

编辑：之前我只用 C++ 和一点 Java 编写代码，所以我以前主要使用数组（在 C++ 中它会像这样：

int main() { 
  int a[100][100];
  for (int i=1, i<=a.length-1, i++) { 
    for (int j=1, i<=a.width-1, j++) { 
      a[i][j] = 1/4 (a[i][j+1]+a[i][j-1]+a[i+1][j]+a[i-1][j]);
    } 
  } 
return 0;
}

Answer 1

In [152]: a = np.arange(6).reshape(2,3)                                                                  
In [153]: a                                                                                              
Out[153]: 
array([[0, 1, 2],
       [3, 4, 5]])

让我们运行你的 nditer 看看它的值：

In [157]: it = np.nditer(a, flags=['multi_index'], op_flags=['readwrite'])                               
In [158]: while not it.finished: 
     ...:     print(it.multi_index, a[it.multi_index], it[0], type(it[0])) 
     ...:     it.iternext() 
     ...:                                                                                                
(0, 0) 0 0 <class 'numpy.ndarray'>
(0, 1) 1 1 <class 'numpy.ndarray'>
(0, 2) 2 2 <class 'numpy.ndarray'>
(1, 0) 3 3 <class 'numpy.ndarray'>
(1, 1) 4 4 <class 'numpy.ndarray'>
(1, 2) 5 5 <class 'numpy.ndarray'>

在每次迭代中，multiindex 是 i,j 索引的元组。 a[it.multiindex] 然后从数组中选择该项目。但是 it[0] 也是那个项目，但包装为 0d 数组。如果您对 0d 数组（形状 ()）的想法不满意，那么 nditer 不适合您（此时）。

如果您只需要顺序索引元组，ndindex 也可以：

In [162]: list(np.ndindex(a.shape))                                                                      
Out[162]: [(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2)]

（事实上，np.lib.index_tricks.py 表明 ndindex 使用 nditer 多重索引。nditer 在 numpy [=64= 中不常用]等级代码。）

或获取指数加值：

In [177]: list(np.ndenumerate(a))                                                                        
Out[177]: [((0, 0), 0), ((0, 1), 1), ((0, 2), 2), ((1, 0), 3), ((1, 1), 4), ((1, 2), 5)]

仅按固定顺序排列的值：

In [178]: a.ravel()                                                                                      
Out[178]: array([0, 1, 2, 3, 4, 5])

但是，在 numpy 中，我们更愿意完全不进行迭代。相反，我们尝试使用快速编译的 numpy 方法编写适用于整个数组的代码。数组迭代很慢，比列表迭代慢。

===

看起来你的迭代，在某种程式化的意义上，是：

for i in range(n):
    for j in range(m):
         a[i,j] = ( a[i,j+1] + a[i,j-1] + a[i+1,j] + a[i-1,j] )/4

有一些细节需要担心。 j+/-1 越界的边缘呢？并且这个计算是连续的，所以 a[i,j] 取决于刚刚对 a[i,j-1] 所做的更改；还是经过缓冲？

在像这样的数组上进行一般顺序迭代计算不适合 numpy。

另一方面，使用 whole-array 切片可以很好地完成缓冲计算

x[1:-1, 1:-1] = (x[:,:-1]+x[:,1:]+x[:-1,:]+x[1:,:])/4

在scipy中还有一些卷积函数对移动进行计算windows。

有没有办法将 multi_index 转换为 int 字符串/列表/数组？

Is there a way to convert a multi_index into a int string / list / array?

python

numpy

multidimensional-array

array-indexing