博客

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是输出的目标数组。其他参数可以参考官方文档。

以下是两个代码实例：

1. 对数组按元素取模运算

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数组。

1. 自定义输出目标数组

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数组的值。