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
来自地板除法的逐元素商。如果 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]))