python - 创建一系列数字随机洗牌的二维数组的有效方法

Question

我需要创建一个由 0 到 N 的随机打乱数字组成的 N x M numpy 数组。也就是说，矩阵的每一行都以随机顺序恰好有 0 到 N 中的每个数字中的一个，没有重复元素，并且每个表示数字。

到目前为止我能想到的最好的是(对于 N = 10 和 M = 5)

import random
import numpy as np

a = np.array([random.sample(range(10), 10) for x in range(5)])

这给了我

[[5 9 1 3 8 2 6 4 0 7]
 [4 8 5 2 9 3 7 6 0 1]
 [8 4 6 7 9 2 0 5 1 3]
 [3 5 4 9 2 0 6 7 1 8]
 [6 0 4 7 3 2 1 8 5 9]]

我当前的解决方案确实有效，但如果可能的话，我只想有一个随机库(我在其他地方使用 numpy，因此仅 numpy 的解决方案是理想的)。我假设 numpy 有一种更简单的方法来做到这一点，但我目前缺少它。

Answer 1

如果问题只是仅使用一个随机库(正如您在评论中所说)，您可以使用 numpy.random.choice使用参数 replace=False，确保每个元素不重复:

import numpy as np

a = np.stack([np.random.choice(range(10), 10, replace=False) for _ in range(5)])

>>> a
array([[0, 9, 3, 5, 8, 7, 1, 4, 6, 2],
       [5, 6, 8, 3, 0, 4, 7, 9, 2, 1],
       [7, 4, 9, 5, 0, 1, 6, 8, 3, 2],
       [9, 0, 3, 8, 5, 7, 6, 1, 4, 2],
       [5, 6, 0, 1, 3, 4, 9, 8, 7, 2]])