Python NumPy delete方法用法及代码示例

例子

从一维数组中删除

考虑以下一维数组：

a = np.array([3,4,5])
a



array([3, 4, 5])

要删除索引 1 处的值：

np.delete(a, 1)



array([3, 5])

从展平的二维数组中删除值

考虑以下二维数组：

a = np.array([[4,5],[6,7]])
a



array([[4, 5],
       [6, 7]])

删除特定索引

要从数组的扁平化版本中删除单个索引：

np.delete(a, 2)



array([4, 5, 7])

删除多个索引

要删除索引 0 和 2 处的值，请传入一个数组：

np.delete(a, [0,2])



array([5, 7])

从二维数组中删除行

假设我们有以下二维数组：

a = np.array([[4,5],[6,7]])
a



array([[4, 5],
       [6, 7]])

删除特定行

要删除行索引 1：

np.delete(a, 1, axis=0)



array([[4, 5]])

删除多行

要删除索引 0 和 2 处的值，请传入一个数组：

np.delete(a, [0,2])



array([5, 7])

从二维数组中删除列

删除特定列

假设我们有以下二维数组：

a = np.array([[4,5],[6,7]])
a



array([[4, 5],
       [6, 7]])

删除列索引 1：

np.delete(a, 1, axis=1)



array([[4],
       [6]])

删除多列

考虑以下二维数组：

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



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

要删除索引 0 和 2 处的列：

np.delete(a, [0,2], axis=1)



array([[2],
       [5],
       [8]])