numpy 中的 multiply 函数用于对数组中的元素进行乘法运算,既可以用于两个数组对应元素相乘,也可用于数组与标量相乘。以下是具体用法和示例:
导入 numpy 库
使用 multiply 函数前需先导入 numpy 库,一般约定俗成的导入方式为:
python
import numpy as np
数组与数组相乘
当 multiply 函数的两个参数都是数组时,它会将两个数组对应位置的元素相乘,生成一个新的数组。两个数组的形状需相同或符合广播机制。例如:
python
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
c = np.multiply(a, b)
print(c)
上述代码中, a 和 b 两个数组对应元素相乘,得到 [4, 10, 18] 。
数组与标量相乘
当 multiply 函数的一个参数是数组,另一个是标量时,它会将数组的每个元素与该标量相乘。例如:
python
a = np.array([1, 2, 3])
scalar = 2
d = np.multiply(a, scalar)
print(d)
代码中,数组 a 的每个元素都与标量 2 相乘,得到 [2, 4, 6] 。