gpt4 book ai didi

python - 无需替换即可生成置换数组

转载 作者:行者123 更新时间:2023-12-05 09:32:57 27 4
gpt4 key购买 nike

我正在尝试获取大小为 (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)

关于python - 无需替换即可生成置换数组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/67616804/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com