Question

给定一个整数k，如何创建一个排列矩阵，其中包含序列1到k的所有可能排列？例如，我们考虑k=2。然后我想创建矩阵：

1 2
2 1

和k=3：

1 1 2 2 3 3
2 3 1 3 1 2 
3 2 3 1 2 1

我尝试过使用numpy.random.permutation，但这只会产生一个排列。所以，我可以继续使用这个函数，附加唯一的排列，直到列数等于k!，但这看起来非常低效。

Answer 1

我的解决方案：

perm_mat = np.zeros(PERM_SIZE,NUM_OF_PERM))
for i in range(NUM_OF_PERM):
perm_mat[:,i] = np.random.permutation(PERM_SIZE))
perm_mat = perm_mat.astype(int)

Answer 2

基于this answer：

import numpy as np
import itertools as it
import math
def myPerms(k):
  f_k=math.factorial(k)
  A=np.empty((k,f_k))
  for i,perm in enumerate(it.permutations(range(k))):
    A[:,i] = perm
  A+=1
  return A

print(myPerms(3))
#[[ 1.  1.  2.  2.  3.  3.]
# [ 2.  3.  1.  3.  1.  2.]
# [ 3.  2.  3.  1.  2.  1.]]

Answer 3

纯粹的numpy解决方案怎么样：

from math import factorial as fac
import numpy as np

def permmat(n):
    if n==1:
        return np.array([[1]], dtype=np.int8)
    fnm1 = fac(n-1)
    pmat_nm1 = permmat(n-1)
    pmat = np.empty((n, fac(n)), dtype=np.int8)
    pmat[0] = np.repeat(np.arange(n,0,-1), fnm1)
    pmat[1:, :fnm1] = pmat_nm1
    for i in range(1,n):
        view = pmat[1:, fnm1*i:fnm1*(i+1)]
        view[:,:] = pmat_nm1
        view[pmat_nm1==(n-i)] = n
    return pmat

print(permmat(4))

输出：

[[4 4 4 4 4 4 3 3 3 3 3 3 2 2 2 2 2 2 1 1 1 1 1 1]
 [3 3 2 2 1 1 4 4 2 2 1 1 3 3 4 4 1 1 3 3 2 2 4 4]
 [2 1 3 1 2 3 2 1 4 1 2 4 4 1 3 1 4 3 2 4 3 4 2 3]
 [1 2 1 3 3 2 1 2 1 4 4 2 1 4 1 3 3 4 4 2 4 3 3 2]]

还有一些性能攻击的空间，但是我懒得写它们。

创建排列矩阵 - numpy

3 个答案: