numpy 删除矩阵中的部分数据 numpy.delete

numpy.delete(arr ,id ,axis=None)
官方链接
功能：
(1)如果输入了axis，则把数组arr在axis指定的维度进行切片，并删除掉id指定下标的元素
(2)如果没有输入axis，则把数组arr扁平化，并删除掉id指定下标的元素

示例：

import numpy as np
mat= np.array([[1,2,3,4], [5,6,7,8], [9,10,11,12]])
print('mat=',mat)
print('删除第0行：',np.delete(mat,0,axis=0))
print('删除第0列：',np.delete(mat,0,axis=1))

结果：

mat= [[ 1  2  3  4]
 [ 5  6  7  8]
 [ 9 10 11 12]]
删除第0行： [[ 5  6  7  8]
 [ 9 10 11 12]]
删除第0列： [[ 2  3  4]
 [ 6  7  8]
 [10 11 12]]