时间戳

首页 > 解决方案 > 使用 numpy 生成数组排列的长列表（重复）

python - 使用 numpy 生成数组排列的长列表（重复）

问题描述

我希望能够生成一个数组列表（任意长度），每个数组都是一个数组 A 的随机排列。

有这个 python 选项：

for x in range(n):
    random.shuffle(A)
    routes.append(A)

但我怀疑使用 numpy 实现这一目标的程序性较少。有什么建议吗？

标签： pythonarraysnumpy

解决方案

考虑这个例子：

In [59]: crr
Out[59]: 
array([[0.83704624, 0.42109715, 0.50779425, 0.93753455, 0.11773652,
        0.08751938],
       [0.83704624, 0.42109715, 0.50779425, 0.93753455, 0.11773652,
        0.08751938],
       [0.83704624, 0.42109715, 0.50779425, 0.93753455, 0.11773652,
        0.08751938],
       [0.83704624, 0.42109715, 0.50779425, 0.93753455, 0.11773652,
        0.08751938],
       [0.83704624, 0.42109715, 0.50779425, 0.93753455, 0.11773652,
        0.08751938]])

In [60]: np.random.shuffle(crr[0])

In [61]: crr
Out[61]: 
array([[0.42109715, 0.50779425, 0.93753455, 0.11773652, 0.08751938,
        0.83704624],
       [0.83704624, 0.42109715, 0.50779425, 0.93753455, 0.11773652,
        0.08751938],
       [0.83704624, 0.42109715, 0.50779425, 0.93753455, 0.11773652,
        0.08751938],
       [0.83704624, 0.42109715, 0.50779425, 0.93753455, 0.11773652,
        0.08751938],
       [0.83704624, 0.42109715, 0.50779425, 0.93753455, 0.11773652,
        0.08751938]])

由于随机洗牌是就地的，在这里我们可以只指定要洗牌的第一行。因此，如果您使用np.repeat或np.tile重复您的 A N 次并将结果重塑为像crr这里这样的二维数组，您可以这样做以达到最终目标：

In [69]: for v in crr:
    ...:     np.random.shuffle(v)

推荐阅读