長さの違う配列を足し算する (python)

計算のイメージ

• 長さが違う配列を用意
  a = [1,2,3,4,5]
  b = [1,2]

---

• bの足りない部分を0で埋めて長さをそろえる
  a = [1,2,3,4,5]
  b = [1,2,0,0,0]

---

• 足し算する
  a + b = [2,4,3,4,5]

やり方

np.padを使って足りない長さを0埋めして計算します．

example.py

import numpy as np

## 長さの違う配列を足し算する
a = np.array([1,2,3,4,5])
b = np.array([1,2])
b_pad = np.pad(b, (0, 3), "constant") #numpy.pad(array, pad_width(before_N, after_N), mode='constant', **kwargs) #

print(a+b_pad) #[2 4 3 4 5]

無事足し算ができます．

np.padの(0, 3)の部分で，右側，左側にどれだけ0を追加するかを指定しています．