引入:
numpy库中最基本也是最重要的对象就是数组。数组带来的一个最明显的好处就是可以进行整体运算。
同形状数组的基本运算:
加法运算:
import numpy as np
a = np.array([1,2,3,4,5])
b = np.array([3,4,5,6,8])
c = a + b
print(c)
减法运算:
import numpy as np
arr1 = arr2 = np.ones((3,4),dtype=int) #创建一个三行四列元素全为1的数组
arr3 = arr1 - arr2
print(arr3)
乘法运算:
import numpy as np
arr1 = np.full((3,3),5.5)
arr2 = np.full((3,3),3)
arr3 = arr1 * arr2
print(arr3)
除法运算:
import numpy as np
arr1 = np.full((3,3),5)
arr2 = np.full((3,3),3)
arr3 = arr1 / arr2
print(arr3)
不同形状数组的相关运算:
加减乘除:
import numpy as np
arr1 = np.array([[1,2,3],[4,5,6]])
arr2 = np.array([6,6,6])
result = arr1 / arr2 #加减乘除都可以
print(result)
这种计算叫做广播运算,arr1的每一行都会对arr2的做关系运算。但广播计算的前提是两个数组的要”形状兼容“,比如每一行的元素都有三个,你也必须乘以三个元素分别能对应。
和普通数值之间的计算:
import numpy as np
arr1 = np.array([[1,2,3],[4,5,6]])
result = arr1 + 1
result1 = arr1 * 2
print(result)
print(result1)
这也是广播计算,普通值会与数组的每个元素进行相应的运算。