numpy的numpy.delete()/insert()/append()函数

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

arr:输入向量

obj:表明哪一个子向量应该被移除。可以为整数或一个int型的向量

axis:表明删除哪个轴的子向量，若默认，则返回一个被拉平的向量

a = np.array(np.arange(12).reshape(3,4))

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

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

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

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

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

注：

numpy.s_[::2]

表示选取奇数。

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]])

3.numpu.append(arr,values,axis=None)

将values插入到目标arr的最后。

注意，这里values跟arr应该为相同维度的向量，例如，如下第一个为错误情况。

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

np.append(a,[1,1,1,1],axis=0)
Traceback (most recent call last):

File "<ipython-input-311-5585d9489cff>", line 1, in <module>
np.append(a,[1,1,1,1],axis=0)

File "D:\Anaconda\lib\site-packages\numpy\lib\function_base.py", line 4586, in append
return concatenate((arr, values), axis=axis)

ValueError: all the input arrays must have same number of dimensions

正确形式为

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]])