numpy.divmod#

numpy.divmod(x1, x2, [out1, out2, ]/, [out=(None, None), ]*, where=True, casting='same_kind', order='K', dtype=None, subok=True[, signature]) = <ufunc 'divmod'>#

同时返回逐元素的商和余数。

版本 1.13.0 中的新功能。

np.divmod(x, y) 等效于 (x // y, x % y),但速度更快,因为它避免了冗余的工作。它用于在 NumPy 数组上实现 Python 内置函数 divmod

参数:
x1array_like

被除数数组。

x2array_like

除数数组。如果 x1.shape != x2.shape,则它们必须能够广播到一个共同的形状(该形状成为输出的形状)。

outndarray、None 或 ndarray 和 None 的元组,可选

结果存储到的位置。如果提供,则其形状必须是输入广播到的形状。如果未提供或为 None,则返回一个新分配的数组。元组(仅可能作为关键字参数)的长度必须等于输出的数量。

wherearray_like,可选

此条件在输入上广播。在条件为 True 的位置,out 数组将设置为 ufunc 结果。在其他位置,out 数组将保留其原始值。请注意,如果通过默认的 out=None 创建了一个未初始化的 out 数组,则其中条件为 False 的位置将保持未初始化。

**kwargs**

对于其他仅限关键字的参数,请参阅 ufunc 文档

返回:
out1ndarray

来自地板除法的逐元素商。如果 x1x2 都是标量,则这是一个标量。

out2ndarray

来自地板除法的逐元素余数。如果 x1x2 都是标量,则这是一个标量。

另请参阅

floor_divide

等效于 Python 的 // 运算符。

remainder

等效于 Python 的 % 运算符。

modf

等效于对于正 xdivmod(x, 1),返回值已切换。

示例

>>> import numpy as np
>>> np.divmod(np.arange(5), 3)
(array([0, 0, 0, 1, 1]), array([0, 1, 2, 0, 1]))

divmod 函数可用作 ndarray 上 np.divmod 的简写。

>>> x = np.arange(5)
>>> divmod(x, 3)
(array([0, 0, 0, 1, 1]), array([0, 1, 2, 0, 1]))