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

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

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

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