輸入：
import numpy as np
import itertools
a = np.array([2, 3, 5, 13, 14, 16, 19, 20, 22, 29, 30, 34, 41, 43, 45, 49, 50,
54, 60, 63, 66, 73, 75, 79, 82, 93]).astype(np.uint8)
b = np.array([ 9, 10, 11, 13, 14, 16, 17, 20, 22, 28, 30, 35, 36, 41, 45, 47, 49,
54, 58, 63, 64, 66, 68, 73, 74, 76, 78, 83, 84, 86, 87, 92, 94, 95]).astype(np.uint8)
c = np.array([ 8, 11, 15, 17, 28, 32, 36, 38, 39, 42, 47, 52, 53, 58, 61, 62, 64,
68, 69, 70, 71, 74, 78, 79, 80, 81, 83, 84, 85, 86, 89, 92]).astype(np.uint8)
d = np.array([ 6, 8, 12, 15, 21, 23, 25, 26, 31, 32, 37, 38, 44, 51, 52, 53, 56,
57, 61, 62, 65, 67, 70, 71, 72, 81, 85, 89, 90]).astype(np.uint8)
e = np.array([ 0, 1, 4, 7, 18, 21, 23, 24, 25, 26, 27, 33, 40, 44, 46, 48, 51,
55, 56, 57, 59, 65, 67, 77, 88, 90, 91]).astype(np.uint8)
當前解決方案：
# Get product and ensure no element is repeated
main_combos = np.array([i for i in itertools.product(a, b, c, d, e) if len(set(i)) == 5])
# Remove duplicates (should not return both [13, 14, 8, 6, 0], [14, 13, 8, 6, 0])
u = np.sort(main_combos, axis=1)
_, idx = np.unique(u, axis=0, return_index=True)
main_combos = main_combos[idx]
這目前在我的機器上需要大約 41 秒才能獲得所有 14,133,909 行，我想知道是否可以提高性能。
每個單獨行的順序需要保持不變，例如每個陣列的第一個元素[2, 9, 8, 6, 0]在最終輸出中的順序應該相同。
uj5u.com熱心網友回復：
您的解決方案在我的機器上需要 58 秒。
如果你不需要使用 Numpy，以下在我的機器上運行 23 秒，使用純 Python
main_combos = {tuple(set(i)): i for i in itertools.product(a, b, c, d, e)}
main_combos = [val for key, val in main_combos.items() if len(key)==5]
轉載請註明出處，本文鏈接：https://www.uj5u.com/yidong/430700.html