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]))