首頁 >後端開發 >Python教學 >如何在 Python 中產生長度為 n 的組合?

如何在 Python 中產生長度為 n 的組合?

Susan Sarandon
Susan Sarandon原創
2024-11-16 04:58:03459瀏覽

How to Generate Combinations of Length n in Python?

長度為 n 的組合

在程式設計中,尋找組合是一項常見任務。組合是從一個較大集合中選擇的不重複的元素集合。例如,從集合[1, 2, 3, 4] 中選擇3 個數字時,可能的組合為:

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

使用itertools 模組

Python itertools 模組提供了一種產生組合的便捷方法。以下程式碼示範如何使用它從數字清單中取得長度為 n 的所有組合:

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)

以上是如何在 Python 中產生長度為 n 的組合?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn