1. 程式人生 > >Python 基礎 排列組合的實現

Python 基礎 排列組合的實現

                     

考慮這樣一個問題,給定一個矩陣(多維陣列,numpy.ndarray()),如何shuffle這個矩陣(也就是對其行進行全排列),如何隨機地選擇其中的k行,這叫組合,實現一種某一維度空間的切片。例如五列中選三列(全部三列的排列數),便從原有的五維空間中降維到三維空間,因為是全部的排列數,故不會漏掉任何一種可能性。

涉及的函式主要有:

  • np.random.permutation()
  • itertools.combinations()
  • itertools.permutations()
# 1.0-5之間的數進行一次全排列>>>np.random.permutation(6)array([3
, 1, 5, 4, 0, 2])# 2. 建立待排矩陣>>>A = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])# 3. shuffle矩陣A>>>p = np.random.permutation(A.shape[0])>>>parray([1, 2, 0])>>>A[p, :]                    array([[ 5,  6,  7,  8],       [ 9, 10, 11, 12],       [ 1,  2,  3,  4]])
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16

C25的實現

>>>from itertools import permutations>>>pertumations(range(5), 2)<itertools.permutations object at 0x0233E360>>>>perms = permutations(range(5), 2)>>>perms[(0, 1), (0, 2), (0, 3), (0, 4), (1, 0), (1, 2), (1, 3), (1, 4), (2, 0), (2, 1), (2, 3
), (2, 4), (3, 0), (3, 1), (3, 2), (3, 4), (4, 0), (4, 1), (4, 2), (4, 3)]>>>len(perms)20
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
# 5. 任取其中的k(k=2)行>>>c = [c for c in combinations(range(A.shape[0]), 2)]>>>A[c[0], :]           # 一種排列array([[1, 2, 3, 4],       [5, 6, 7, 8]])
  • 1
  • 2
  • 3
  • 4
  • 5