2014-12-09 2 views
0

У меня есть два Numpy массива, позволяет сказатьдобавить только определенные colums в массиве - питон

A = array([ [a11, a12, a13], [a21, a22, a23], [a31, a32, a33] ]) 
B = array([ [b11, b12, b13], [b21, b22, b23], [b31, b32, b33] ]) 

и я хочу, чтобы получить (Edited)

C = array([ [a11, a12+b12, a13], [a21, a22+b22, a23], [a31, a32+b32, a33] ]) 

я мог бы добиться этого с для цикла, но я считаю, что в Python должен быть более элегантный способ.

ответ

2

Вы можете не просто сделать следующее:

C = A.copy() 
C[1] += B[1] # (or appropriate indexes) 

OK - отредактированный решение:

C = A.copy() 
C[:,1] += B[:,1] 
+0

жаль, что я написал массив C неправильно. Предполагается, что это C = массив ([[a11, a12 + b12, a13], [a21, a22 + b22, a23], [a31, a32 + b32, a33]]) – Tom83B

+0

, но я вижу, что ваш ответ в любом случае правильный - Я могу просто использовать C = A.copy() C [:, 1] + = B [:, 1] отлично работает – Tom83B