我正在尝试获取大小为 (1000, 7) 的矩阵，其中每一行都是数字 0 到 6 的随机排列。 我正在使用以下代码来实现需求。

import numpy as np

permute_array = []
for i in range(1000):
    permute_array.append(np.random.permutation(7))
permute_array = np.array(permute_array)

这很慢，因为 for 循环需要相当长的时间才能运行。是否有生成此数组的一次性方法？谢谢。

最佳答案

import numpy as np
permute_array = np.argsort(np.random.rand(1000, 7), axis=1)

https://stackoverflow.com/questions/67616804/