numpy删除，插入，增加元素

删除

numpy.delete(arr,obj,axis=None) 

arr:输入向量
obj:表明哪一个子向量应该被移除。可以为整数或一个int型的向量
axis:表明删除哪个轴的子向量，若默认，则返回一个被拉平的向量

a = np.array(np.arange(12).reshape(3,4))
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

np.delete(a,1,0)
array([[ 0,  1,  2,  3],
       [ 8,  9, 10, 11]])

np.delete(a,1,1)
array([[ 0,  2,  3],
       [ 4,  6,  7],
       [ 8, 10, 11]])

np.delete(a,[0,1],1)
array([[ 2,  3],
       [ 6,  7],
       [10, 11]])

np.delete(a,np.s_[::2],1)
array([[ 1,  3],
       [ 5,  7],
       [ 9, 11]])

numpy.s_[::2]表示选取奇数

插入

numpy.insert(arr,obj,value,axis=None) 

value为插入的数值
arr:为目标向量
obj:为目标位置
value:为想要插入的数值
axis:为插入的维度

np.insert(a,1,[1,1,1,1],0)
Out[309]: 
array([[ 0,  1,  2,  3],
       [ 1,  1,  1,  1],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

append
将values插入到目标arr的最后。
注意，这里values跟arr应该为相同维度的向量，例如，如下第一个为错误情况。

a
Out[312]: 
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

np.append(a,[[1,1,1,1]],axis=0)
Out[313]: 
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [ 1,  1,  1,  1]])

查询
有两种方式
一种就是直接用切片的方式

a[:,:2]
array([[0, 1],
       [4, 5],
       [8, 9]])

第二种就是用where

np.where(a>2)
(array([0, 1, 1, 1, 1, 2, 2, 2, 2]), array([3, 0, 1, 2, 3, 0, 1, 2, 3]))

上面返回的是坐标集合，第一个是x，第二个是y