[PYTHON] Versuchen Sie, NumPys add.at mit Julia zu reproduzieren

Der Titel wie er ist.

NumPy-Code, den Sie reproduzieren möchten

Dies ist eine direkte Änderung in NumPy.

Python(NumPy)

>>> A = np.ones((3,3))
>>> A
array([[1., 1., 1.],
       [1., 1., 1.],
       [1., 1., 1.]])
>>> B = np.array([[1, 1, 1], [2, 2, 2]])
>>> B
array([[1, 1, 1],
       [2, 2, 2]])
>>> np.add.at(A, [0, 2], B)
>>> A
array([[2., 2., 2.],
       [1., 1., 1.],
       [3., 3., 3.]])

Julia Code

Beachten Sie, dass der Operator . + Zum Operator + = hinzugefügt wird, um ihn zu . + = Zu machen, und dass eine Übertragung durchgeführt wird. + = sendet nicht und ist keine direkte Änderung. Allerdings sendet . + = (Natürlich) Sendungen, aber es handelt sich um eine ** In-Place-Änderung ** [^ 1].

Julia

julia> A = ones(3,3)
3×3 Array{Float64,2}:
 1.0  1.0  1.0
 1.0  1.0  1.0
 1.0  1.0  1.0

julia> B = [1. 1. 1.; 2. 2. 2.]
2×3 Array{Float64,2}:
 1.0  1.0  1.0
 2.0  2.0  2.0

julia> selectdim(A, 1, [1, 3]) .+= B
2×3 view(::Array{Float64,2}, [1, 3], :) with eltype Float64:
 2.0  2.0  2.0
 3.0  3.0  3.0

julia> A
3×3 Array{Float64,2}:
 2.0  2.0  2.0
 1.0  1.0  1.0
 3.0  3.0  3.0

Sources numpy.ufunc.at — NumPy v1.19 Manual Arrays · The Julia Language Multi-dimensional Arrays · The Julia Language Mathematical Operations and Elementary Functions · The Julia Language