在 NumPy 中刪除行

本教程將介紹如何在 Python 中從多維 NumPy 陣列中刪除一行。

使用 numpy.delete() 函式刪除 NumPy 行

如果我們有一個多維 NumPy 陣列並想從中刪除特定行，我們可以使用 numpy.delete() 函式。numpy.delete() 函式 從 NumPy 陣列中沿指定軸刪除特定索引處的條目。numpy.delete() 函式將陣列、要刪除的索引和要刪除的軸作為引數，並返回一個子陣列，其中刪除了指定索引和指定軸。下面的程式碼示例向我們展示瞭如何使用 numpy.delete() 函式從多維陣列中刪除一行。

import numpy as np

array = np.array([[1,2,3],[4,5,6],[7,8,9]])

array = np.delete(array,(1), axis = 0)
print(array)

輸出：

[[1 2 3]
 [7 8 9]]

我們使用上面程式碼中的 np.delete() 函式從二維 NumPy 陣列 array 中刪除了第二行。我們首先使用 np.array() 函式建立了二維 NumPy 陣列 array。然後我們沿著軸 0 刪除索引 1 處的條目，這是 array 的第二行，並將結果儲存在 array 中。