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

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11
• 12
• 13
• 14
• 15
• 16
• 17
• 18
• 19
• 20
• 21
• 22
• 23
• 24
• 25
• 26
• 27
• 28
• 29
• 30

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

• 1
• 2
• 3
• 4
• 5
• 6

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

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11
• 12
• 13
• 14
• 15
• 16

正确形式为

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