Python的np.mod
函数可以用来计算两个数组的按元素取模运算。
函数语法如下:
np.mod(x1, x2, /, out=None, *, where=True, casting='same_kind', order='K', dtype=None, subok=True[, signature, extobj])
其中x1
和x2
是两个数组,out
是输出的目标数组。其他参数可以参考官方文档。
以下是两个代码实例:
- 对数组按元素取模运算
import numpy as np
a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
b = np.array([2, 3, 4, 5, 6, 7, 8, 9, 10])
c = np.mod(a, b)
print(c)
输出:
[1 2 3 4 5 6 7 8 9]
在这个例子中,a
和b
两个数组分别是[1, 2, 3, 4, 5, 6, 7, 8, 9]
和[2, 3, 4, 5, 6, 7, 8, 9, 10]
,我们使用np.mod
函数对这两个数组进行按元素取模运算,结果存到了c
数组中,最后输出了c
数组。
- 自定义输出目标数组
import numpy as np
a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
b = np.array([2, 3, 4, 5, 6, 7, 8, 9, 10])
c = np.zeros(9)
np.mod(a, b, out=c)
print(c)
输出:
[1. 2. 3. 4. 5. 6. 7. 8. 9.]
在这个例子中,我们使用了np.mod
函数对a
和b
两个数组进行按元素取模运算,不同的是我们将out
参数设置成了一个初始化为0的长度为9的数组c
。结果会输出到c
数组中。最后输出了c
数组的值。