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'>#

同时返回按元素的商和余数.

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

参数:
x1array_like

被除数数组.

x2array_like

除数数组.如果 x1.shape != x2.shape ,则它们必须可以广播到通用形状(这将成为输出的形状).

outndarray, None, or tuple of ndarray and None, optional

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

其中类数组,可选

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

\kwargs

对于其他仅限关键字的参数,请参见 ufunc docs .

返回:
out1ndarray

由 floor division 产生的按元素的商.如果 x1x2 都是标量,则这是一个标量.

out2ndarray

来自 floor division 的按元素的余数.如果 x1x2 都是标量,则这是一个标量.

参见

floor_divide

等效于 Python 的 // 运算符.

remainder

等效于 Python 的 % 运算符.

modf

对于正数 x ,等效于切换了返回值的 divmod(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]))