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

来源:互联网 发布:java char数组倒序输出 编辑:程序博客网 时间:2024/05/14 02:29

1.numpy.delete(arr,obj,axis=None)
arr:输入向量
obj:表明哪一个子向量应该被移除。可以为整数或一个int型的向量
axis:表明删除哪个轴的子向量,若默认,则返回一个被拉平的向量

a = np.array(np.arange(12).reshape(3,4))aOut[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应该为相同维度的向量,例如,如下第一个为错误情况。

aOut[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

正确形式为

aOut[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]])
原创粉丝点击