首頁 >後端開發 >Python教學 >如何在 Python 中有效地展平嵌套列表?

如何在 Python 中有效地展平嵌套列表?

Susan Sarandon
Susan Sarandon原創
2024-12-31 03:09:10438瀏覽

How Can I Efficiently Flatten a Nested List in Python?

展平列表列表

問題陳述:

你有一個嵌套列表,需要將其展平為一個列表。例如,您有一個列表列表,例如:

[[1, 2, 3],
[4, 5, 6],
[7],
[8, 9]]

目標是將這個結構展平為單一列表:

[1, 2, 3, 4, 5, 6, 7, 8, 9]

解決方案:

要展平列表列表,您可以使用巢狀列表來理解。此技術涉及迭代每個內部列表(在程式碼中以 xs 表示)並將每個單獨的元素(由 x 表示)附加到展平列表(名為flat_list):

flat_list = [
    x
    for xs in xss
    for x in xs
]

這相當於以下內容使用巢狀循環明確迭代:

flat_list = []

for xs in xss:
    for x in xs:
        flat_list.append(x)

或者,您可以建立一個函數來執行此扁平化操作:

def flatten(xss):
    return [x for xs in xss for x in xs]

使用此函數,您可以如下方式展平輸入清單:

flat_list = flatten([[1, 2, 3],
                   [4, 5, 6],
                   [7],
                   [8, 9]])

這種方法效率很高,且效能明顯優於其他方法,例如使用或減少。列表理解技術確保每個元素僅複製到展平列表一次。

以上是如何在 Python 中有效地展平嵌套列表?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

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