Python numpy insert()、 delete()、append()函数的用法

简介：

这三个函数的功能是增删矩阵或数组的某一行/列，接下来逐个介绍函数用法。

import numpy as np

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

一、numpy.insert()

• 函数语法：

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

• 函数作用：
  将向量插入某一行或列
• 参数说明：
  arr：array
        输入矩阵
  obj：int
        插入在第几行/列之前
  values：array
        要插入的矩阵
  axis：int
        插入某一行(0)还是列(1)
• 返回值：
  返回一个插入向量后的数组。若axis=None，则返回一个扁平(flatten)数组
• 例子：

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

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

二、numpy.delete()

• 函数语法：

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

• 函数作用：
  删除某一行或列
• 参数说明：
  arr：array
        输入矩阵
  obj：int
        删除哪一行或列
  axis：int
        删除行(0)还是列(1)
• 返回值：
        返回删除某一行/列的数组
• 例子：

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

>>> np.delete(a,2,0)
array([[0, 1, 2, 3],
       [4, 5, 6, 7]])

三、numpy.append()

• 函数语法：

numpy.append(arr, values, axis=None)

• 函数作用:
  在数组末尾添加一行/列数组
• 参数说明:
  arr：array
        输入矩阵
  values：array
        要插入的矩阵
  axis：int
        按着行(0)还是列(1)插入
• 例子:

>>> np.append(a, [[1,2,3,4]], 0) #这里注意插入的形式
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [ 1,  2,  3,  4]])

>>> np.append(a,[[1],[2],[3]], 1)
array([[ 0,  1,  2,  3,  1],
       [ 4,  5,  6,  7,  2],
       [ 8,  9, 10, 11,  3]])