要素ごとのnumpy関数の引数として（行、列）を追加しますか？

Question

Numpyは非常に自然に要素単位の操作を行います。私はちょうどループのネストされたを作成することができ

def neighbor_count(arr, row, col): 
hood = arr[row-1:row+1, col-1, col+1] 
return np.sum(hood)-arr[row,col] 

：

for (i, row) in enumerate(arr[1:-1]): 
for (j, col) in enumerate (row[1:-1]): 
    c = neighbor_count(arr, i+1, j+1) 

しかし、それは諦めと行くような感じしかし、多くの場合、私のような、機能を書くときも、要素の位置を持ちたいです私はC++でどのように働くかに戻る。

賢明なPythonicソリューションはありますか？

Answer 1

あなたはenumerateと同じように機能しますが、ndarray構造の場合はnumpy.ndenumerateが必要です。

結果の値をどのように組み合わせようとしているのか完全には分かっていないため、特定のユースケースに正確な答えはありませんが、基本的な考え方はこれです。 ndenumerateはイテレータを返すので、あなたがそれに取り組む必要があるために使用しているすべてのitertools機能することを

for (r, c), i in np.ndenumerate(arr): 
    do_something() 

注意。