只添加数组中的特定列 - python

时间:2014-12-09 16:30:32

标签: python arrays numpy

我有两个numpy数组,让我们说

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] ])

我可以通过for循环实现这一点,但我相信在Python中必须有一种更优雅的方式。

1 个答案:

答案 0 :(得分:2)

你能不能简单地做到以下几点:

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

确定 - 已编辑的解决方案:

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