Numpy常用random函数
Numpy数组创建函数
import numpy as np
# 生成全是0的3X3的矩阵
nd_zero = np.zeros([3,3])
#生成与nd_zero形状一样全0矩阵
np.zeros_like(nd_zero)
#生成全是1的3X3的矩阵
nd_one = np.ones([3,3])
# 生成3阶的单位矩阵
np_three = np.eye(3)
#生成3阶对角矩阵
nd_three_j = np.diag([1,2,4])
print(nd_zero)
print('============')
print(nd_one)
print('============')
print(np_three)
print('============')
print(nd_three_j)
运行结果如下:
有时候生成的数据要进行存储,代码如下:
import numpy as np
nd_three_j = np.diag([1,2,4])
#数据保存
np.savetxt(X=nd_three_j,fname='./np_date1.txt')
print('================')
#数据读取
nd_read = np.loadtxt('./np_date1.txt')
print(nd_read)
运行结果如下:
利用arange、linspace函数生成数组
arange是numpy模块中的函数,与python的内置函数range的功能类似。
例:
import numpy as np
print(np.arange(0,10))
print(np.arange(9,-1,-1))
运行结果:
linspace也是numpy模块中常用的函数
格式为:np.linspace(start,stop,num=10,endpoint=True,retstep=False,dtype=None)
endpoint(包含终点,默认为True)
retstep(设置为True,会返回一个带步长的ndarray)
import numpy as np
print(np.linspace(0,0.5,5))
print(np.linspace(0,0.5,5,endpoint=False))
print(np.linspace(0,0.5,5,retstep=True))
运行结果为: