@R星校长
第3
关:基础操作
算术运算
如果想要对ndarray
对象中的元素做elementwise
(逐个元素地)的算术运算非常简单,加减乘除即可。代码如下:
import numpy as np
a = np.array([0, 1, 2, 3])
# a中的所有元素都加2,结果为[2, 3, 4, 5]
b = a + 2
# a中的所有元素都减2,结果为[-2, -1, 0, 1]
c = a - 2
# a中的所有元素都乘以2,结果为[0, 2, 4, 6]
d = a * 2
# a中的所有元素都平方,结果为[0, 1, 4, 9]
e = a ** 2
# a中的所有元素都除以2,结果为[0, 0.5, 1, 1.5]
f = a / 2
# a中的所有元素都与2比,结果为[True, True, False, False]
g = a < 2
矩阵运算
相同shape
的矩阵A
与矩阵B
之间想要做elementwise
运算也很简单,加减乘除即可。代码如下:
import numpy as np
a = np.array([[0, 1], [2, 3]])
b = np.array([[1, 1], [3, 2]]