删除
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