Python 是一門廣泛應用于各種領域的編程語言,它具有簡潔的語法、靈活的數(shù)據(jù)類型和強大的庫函數(shù)。在 Python 中處理數(shù)組是一項常見的任務,其中之一是展平數(shù)組,在本文中我們將會介紹兩種顯示數(shù)組的方法。
首先,我們來看第一種方法:
import itertools arr = [[1, 2], [3, 4], [5, 6]] # 使用 itertools.chain() 函數(shù)展平數(shù)組 flattened_arr = list(itertools.chain(*arr)) print(flattened_arr)
在這個示例代碼中,首先我們導入 Python 的內(nèi)置庫函數(shù) itertools。然后我們創(chuàng)建了一個由多個子數(shù)組組成的數(shù)組(這個數(shù)組的維度是 3×2),接下來我們調(diào)用 itertools.chain() 函數(shù)并將數(shù)組通過 * 運算符展開,這樣我們就能得到一個展平的數(shù)組。最后,我們在命令行中打印了展平的數(shù)組,輸出結果為 [1, 2, 3, 4, 5, 6]。
接下來,我們來看第二種方法:
import numpy as np arr = [[1, 2], [3, 4], [5, 6]] # 使用 Numpy 庫的 flatten() 函數(shù)展平數(shù)組 flattened_arr = np.array(arr).flatten().tolist() print(flattened_arr)
在這個示例代碼中,我們導入了 Numpy 庫,這個庫是 Python 的一個核心科學計算庫,它支持大量數(shù)組操作及數(shù)學函數(shù)。然后我們創(chuàng)建了一個與第一種方法相同的數(shù)組,并調(diào)用了 Numpy 的 flatten() 函數(shù)來展平這個數(shù)組。最后,我們使用 .tolist() 函數(shù)將得到的展平數(shù)組轉換為 Python 的列表格式,并打印到命令行中,輸出結果依然是 [1, 2, 3, 4, 5, 6]。
通過以上兩種方法,我們可以快速展平一個多維的 Python 數(shù)組,這對于處理數(shù)據(jù)文件或進行數(shù)據(jù)分析等任務非常有用。