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
。- 参数:
- x1类数组
被除数数组。
- x2类数组
除数数组。如果
x1.shape != x2.shape
,它们必须能够广播到共同的形状(这将成为输出的形状)。- outndarray、None 或 ndarray 和 None 的元组,可选
结果存储的位置。如果提供,它必须具有输入可广播到的形状。如果未提供或为 None,则返回一个新分配的数组。元组(仅作为关键字参数可能)的长度必须等于输出的数量。
- where类数组,可选
此条件会广播到输入。在条件为 True 的位置,out 数组将被设置为 ufunc 结果。在其他位置,out 数组将保留其原始值。请注意,如果通过默认的
out=None
创建未初始化的 out 数组,则其中条件为 False 的位置将保持未初始化。- **kwargs
有关其他仅限关键字的参数,请参见ufunc 文档。
- 返回:
- out1ndarray
地板除法产生的逐元素商。如果 x1 和 x2 都是标量,则这是一个标量。
- out2ndarray
地板除法产生的逐元素余数。如果 x1 和 x2 都是标量,则这是一个标量。
另请参见
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]))