首頁 >後端開發 >Python教學 >如何在 Python 中從列表推導式展平嵌套列表?

如何在 Python 中從列表推導式展平嵌套列表?

DDD
DDD原創
2024-11-14 11:07:02895瀏覽

How to Flatten Nested Lists from List Comprehensions in Python?

扁平化列表理解的結果

在Python 中,使用列表理解將函數應用於列表的每個元素可能會導致嵌套列表。例如,考慮一個列表 A 和一個將 A 的每個元素轉換為列表的函數 f,列表推導式 [f(a) for a in A] 會產生一個列表列表。

要取得一個扁平化的列表,與函數式程式設計的「flatmap」或.NET的「SelectMany」運算子不同,可以在列表中引入嵌套理解:

<code class="python">[filename for path in dirs for filename in os.listdir(path)]</code>

這在功能上等同於以下使用多個嵌套循環的程式碼:

<code class="python">filenames = []
for path in dirs:
    for filename in os.listdir(path):
        filenames.append(filename)</code>

採用此方法可確保產生平面列表,其中元素透過以下方式取得首先迭代外部列表dirs,然後將函數f 傳回的內部列表套用至dirs 中的每個元素。

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

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