Numpy打乱数组或打乱矩阵行

使用numpy.random.shuffle函数,能够打乱ndarray对象的第一维度,对于数组来说,就是整体被打乱。对于矩阵来说,第一维度行被打乱。可以在打乱训练数据或测试模型性能的时候使用。

  • Parameters: x: array_like
  • Returns: None

e.g.

>>> arr = np.arange(10)
>>> np.random.shuffle(arr)
>>> arr
[9, 1, 2, 7, 5, 3, 0, 8, 4, 6]

多维数组

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

参考:

  1. numpy.random.shuffle打乱顺序函数
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐