python np.mod函数怎么用?

  • Post category:Python

Python的np.mod函数可以用来计算两个数组的按元素取模运算。

函数语法如下:

np.mod(x1, x2, /, out=None, *, where=True, casting='same_kind', order='K', dtype=None, subok=True[, signature, extobj])

其中x1x2是两个数组,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]

在这个例子中,ab两个数组分别是[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函数对ab两个数组进行按元素取模运算,不同的是我们将out参数设置成了一个初始化为0的长度为9的数组c。结果会输出到c数组中。最后输出了c数组的值。