一、背景知识
以往用python创建二维数组/列表,可以这样写:
2d_array = [[0 for col in range(cols)] for row in range(rows)]
但是如果要创建三维,n维数组怎么办呢?
这时候就需要用到今天学的Numpy这个库来创建数组
比如用Numpy中的arange方法快速创建一个一维数组:
import numpy as np #引入numpy库并取别名为np(取别名方便之后编写)
a = np.arange(10) #指将数值0 1 2 3 ...9赋值给a这个变量,即一个一维数组
创建多维数组的方法:
创建一个2*6的二维数组:
import numpy as np
b = np.array([np.arange(6),np.arange(6)])
创建一个4*5的二维数组:
import numpy as np
x = [y for y in range(6)]
b=np.array([x]*4)
二、基本运算
Numpy库可以直接进行一些四则运算,快速的处理两个Numpy数组
初始a , b数组
a = np.array([[1,2,3],[4,5,6]])
b = np.array([[4,5,6],[1,2,3]])
- 加法
输入In:np.add(a,b) 或 a+b
输出Out:array([[5, 7, 9],
[5, 7, 9]])
- 减法
In:np.subtract(a,b) 或 a-b
Out:array([[-3, -3, -3],
[ 3, 3, 3]])