numpy 实现两个不同维度的数组相加

import numpy as np
a = np.arange(9).reshape(3,3)
b = np.zeros([2,2], dtype=int) - 100
print(a)
print()
print(b)
a[1:3, :2] += b
print(a)

结果:
a: [[0   1   2]
   [3   4   5]
   [6   7   8]]

b: [[-100   -100]
   [-100   -100]]

a[1:3, :2] += b :   [[   0,       1,      2],
             [-97,   -96,      5],
             [-94,   -93,      8]]

posted @ 2021-04-30 10:37  id_ning  阅读(1697)  评论(0)    收藏  举报