📅  最后修改于: 2023-12-03 15:36:32.461000             🧑  作者: Mango
在 NumPy 中,我们可以使用 numpy.divide()
函数将数组的每一行除以一个向量元素。该函数的语法如下:
numpy.divide(x1, x2, out=None, where=True, casting='same_kind', **kwargs)
其中 x1
为被除数,x2
为除数。
示例代码如下:
import numpy as np
# 定义数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 定义向量
vec = [1, 2, 3]
# 将数组的每一行除以向量的对应元素
result = np.divide(arr, vec)
print(result)
输出结果:
array([[1. , 1. , 1. ],
[4. , 2.5 , 2. ],
[7. , 4. , 3. ]])
可以看到,数组的每一行都被对应元素除以了向量。这是因为 NumPy 自动对齐了相同位置的元素。
如果被除数和除数的形状不相同,NumPy 会报错。因此,在进行除法操作之前,我们需要确保数组和向量的形状相同。
除此之外,numpy.divide()
还支持选项参数 out
、where
和 casting
。其中,out
指定输出结果存放的位置,where
指定进行运算的位置,casting
指定运算时的数据类型。如果不需要用到这些选项,可以直接忽略它们。