如何有效地找到 NumPy 數組中的唯一行？

在NumPy 數組中找到唯一行：一種高效的解決方案

在數據分析和處理中，經常需要從中提取唯一值給定的資料集。在這種情況下，讓我們考慮在 NumPy 陣列中尋找唯一行的問題。

目標：

給定一個 NumPy 數組，目標是識別並取得僅包含原始數組中唯一行的數組。

高效解決方案：

從 NumPy 版本 1.13 開始，引入了一種用於查找唯一行的有效解決方案。透過利用 np.unique 函數並指定 axis 參數，我們可以輕鬆實現這一點：

unique_rows = np.unique(original_array, axis=0)

透過將 axis 參數設為 0，我們指示 NumPy 獨立分析原始陣列的每一行。此操作按元素比較行並傳回僅包含唯一行的新陣列。

範例：

考慮下列NumPy 陣列a：

a = np.array([[1, 1, 1, 0, 0, 0],
               [0, 1, 1, 1, 0, 0],
               [0, 1, 1, 1, 0, 0],
               [1, 1, 1, 0, 0, 0],
               [1, 1, 1, 1, 1, 0]])

要從a 取得唯一行，我們可以使用下列指令code:

unique_rows = np.unique(a, axis=0)

這將產生一個新的陣列unique_rows，其中包含以下行：

unique_rows = np.array([[1, 1, 1, 0, 0, 0],
                         [0, 1, 1, 1, 0, 0],
                         [1, 1, 1, 1, 1, 0]])

我們可以看到，原始數組中的唯一行已被成功提取。

以上是如何有效地找到 NumPy 數組中的唯一行？的詳細內容。更多資訊請關注PHP中文網其他相關文章！