简介:
这三个函数的功能是增删矩阵或数组的某一行/列,接下来逐个介绍函数用法。
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]])