Python numpy divmod用法及代码示例

本文简要介绍 python 语言中 numpy.divmod 的用法。

用法:

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

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

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

参数：

x1： array_like

股息数组。

x2： array_like

除数数组。如果 x1.shape != x2.shape ，它们必须可以广播到一个公共形状(成为输出的形状)。

out： ndarray，None，或 ndarray 和 None 的元组，可选

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

where： 数组，可选

此条件通过输入广播。在条件为真的位置，out数组将设置为 ufunc 结果。在其他地方，out数组将保留其原始值。请注意，如果未初始化out数组是通过默认创建的out=None，其中条件为 False 的位置将保持未初始化状态。

**kwargs：

对于其他仅关键字参数，请参阅 ufunc 文档。

返回：

out1： ndarray

由 floor 除法产生的逐元素商。如果 x1 和 x2 都是标量，则这是一个标量。

out2： ndarray

楼层除法的元素余数。如果 x1 和 x2 都是标量，则这是一个标量。

例子：

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

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

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