Python numpy Generator.permutation用法及代碼示例

本文簡要介紹 python 語言中 numpy.random.Generator.permutation 的用法。

用法:

random.Generator.permutation(x, axis=0)

隨機排列一個序列，或返回一個排列後的範圍。

參數：

x： int 或 數組

如果x是一個整數，隨機排列np.arange(x).如果x是一個數組，製作一個副本並隨機打亂元素。

axis： 整數，可選

x 被打亂的軸。默認值為 0。

返回：

out： ndarray

排列的序列或數組範圍。

例子：

>>> rng = np.random.default_rng()
>>> rng.permutation(10)
array([1, 7, 4, 3, 0, 9, 2, 5, 8, 6]) # random

>>> rng.permutation([1, 4, 9, 12, 15])
array([15,  1,  9,  4, 12]) # random

>>> arr = np.arange(9).reshape((3, 3))
>>> rng.permutation(arr)
array([[6, 7, 8], # random
       [0, 1, 2],
       [3, 4, 5]])

>>> rng.permutation("abc")
Traceback (most recent call last):
    ...
numpy.AxisError: axis 0 is out of bounds for array of dimension 0

>>> arr = np.arange(9).reshape((3, 3))
>>> rng.permutation(arr, axis=1)
array([[0, 2, 1], # random
       [3, 5, 4],
       [6, 8, 7]])