Atribuindo índices de matriz idênticos de uma só vez em Python / Numpy
Eu quero encontrar uma maneira rápida (sem loop) em Python para atribuir índices de reoccuração de uma matriz. Este é o resultado desejado usando um loop for:
import numpy as np
a=np.arange(9, dtype=np.float64).reshape((3,3))
# The array indices: [2,3,4] are identical.
Px = np.uint64(np.array([0,1,1,1,2]))
Py = np.uint64(np.array([0,0,0,0,0]))
# The array to be added at the array indices (may also contain random numbers).
x = np.array([.1,.1,.1,.1,.1])
for m in np.arange(len(x)):
a[Px[m]][Py[m]] += x
print a
%[[ 0.1 1. 2.]
%[ 3.3 4. 5.]
%[ 6.1 7. 8.]]
Quando tento adicionarx
paraa
nos índicesPx,Py
Obviamente, não obtenho o mesmo resultado (3,3 vs. 3,1):
a[Px,Py] += x
print a
%[[ 0.1 1. 2.]
%[ 3.1 4. 5.]
%[ 6.1 7. 8.]]
Existe uma maneira de fazer isso com numpy? Obrigado.