(numpy)基木運用-3

向量化運算

相加

import numpy as np

a=np.array([[1,2,3],[4,5,6]])

b=np.array([1,2,3])

c=a+b

print(c)

[[2 4 6]

 [5 7 9]]

相減

a=np.array([[1,2,3],[4,5,6]])

b=np.array([1,2,3])

c=a-b

print(c)

[[0 0 0]

 [3 3 3]]

相乘

a=np.array([[1,2,3],[4,5,6]])

b=np.array([1,2,3])

c=a*b

print(c)

[[ 1  4  9]

 [ 4 10 18]]

相除

a=np.array([[1,2,3],[4,5,6]])

b=np.array([1,2,3])

c=a/b

print(c)

[[1.  1.  1. ]

 [4.  2.5 2. ]]

數組廣播

import numpy as np

a=np.array([[0],[1],[2],[3]])

b=np.array([1,2,3])

print(a)

print(b)

print(a.shape)

print(b.shape)

[[0]

 [1]

 [2]

 [3]]

[1 2 3]

(4, 1)

(3,)

可以看到上面a是4欄1列

b是3欄

以圖例表示上面運算

a=np.array([[0],[1],[2],[3]])

b=np.array([1,2,3])

print(a+b)

[[1 2 3]

 [2 3 4]

 [3 4 5]

 [4 5 6]]

圖像化之後就比較清楚這些二維陣列的排列運作方式了