如何使用 itertools 檢索指定長度的組合
在程式設計中,可能需要從給定清單產生組合。組合是從集合或清單中選擇的元素,其中元素的順序很重要。
考慮從數字清單中檢索長度為 n 的所有組合的任務。例如,列表[1, 2, 3, 4] 且n 設定為3,則預期出現以下組合:
[1, 2, 3]
[1, 2, 4]
[1, 3, 4]
[2, 3, 4]
要使用Python 的itertools 模組實現此目的,可以使用以下程式碼:
import itertools for comb in itertools.combinations([1, 2, 3, 4], 3): print(comb)
此程式碼將依需求產生組合:
(1, 2, 3)
(1, 2, 4)
(1, 3, 4)
(2, 3, 4)
如上所述,itertools.combinations 提供了一種高效且簡單的解決方案,用於從給定清單中檢索指定長度的組合。
以上是如何使用 itertools 從清單中產生特定長度的組合?的詳細內容。更多資訊請關注PHP中文網其他相關文章!