Python numpy库㈢
前言
一、numpy库在线性代数的应用
import numpy as np
Array1 = np.arange(6).reshape((2,3))
Array1
#array([[0, 1, 2],
# [3, 4, 5]])
Array2 = np.arange(6).reshape((3,2))
Array2
#array([[0, 1],
# [2, 3],
# [4, 5]])
Array1.dot(Array2) # 矩阵內积
#array([[10, 13],
# [28, 40]])
Array3 = np.arange(4).reshape((2,2))
Array3
#array([[0, 1],
# [2, 3]])
detArray3 = np.linalg.det(Array3) # 计算Array3的行列式
detArray3
#-2.0
invArray3 = np.linalg.inv(Array3) # 计算Array3的逆矩阵
invArray3
#array([[-1.5, 0.5],
# [ 1. , 0. ]])
eyeArray=Array3.dot(invArray3) # 互逆矩阵进行內积运算,验证结果
eyeArray
#array([[1., 0.],
# [0., 1.]])
鸡兔同笼问题
solve()函数可以求解形如Ax=b的线性方程组,其中A为系数矩阵,b为一维或者二维的数组,x是未知变量
import numpy as np
heads,foots = 35,94
A = np.array([[1,1],[2,4]]) #方程组的系数矩阵
b = np.array([heads,foots]) #方程组右侧的常数矩阵
x = np.linalg.solve(A,b) #solve()函数返回方程组的解
print("鸡:{},兔:{}".format(x[0],x[1]))
二、多项式的应用
(5)polyadd(p1,p2)函数用于多项式求和。
p1 = np.poly1d(np.array([1,2,3]))
p2 = np.poly1d(np.array([1,2,3,4]))
print(p1)
print(p2)
print(p1+p2) #直接使用"+"运算符
print(np.polyadd(p1,p2)) #使用polyadd()函数实现多项式加法
输出:
2
1 x + 2 x + 3
3 2
1 x + 2 x + 3 x + 4
3 2
1 x + 3 x + 5 x + 7
3 2
1 x + 3 x + 5 x + 7
(6)polysub(p1,p2)函数用于多项式求差。
p1 = np.poly1d(np.array([1,2,3]))
p2 = np.poly1d(np.array([1,2,3,4]))
print(p2-p1)
print(np.polysub(p2,p1))
输出:
3 2
1 x + 1 x + 1 x + 1
3 2
1 x + 1 x + 1 x + 1
(7)polymul(p1,p2)函数用于多项式求积。
p1 = np.poly1d(np.array([1,2,3]))
p2 = np.poly1d(np.array([1,2,3,4]))
print(p1 * p2)
print(np.polymul(p1,p2))
输出:
5 4 3 2
1 x + 4 x + 10 x + 16 x + 17 x + 12
5 4 3 2
1 x + 4 x + 10 x + 16 x + 17 x + 12
(8)polydiv(p1,p2)函数用于多项式求商。
p1 = np.poly1d(np.array([1,2,3]))
p2 = np.poly1d(np.array([1,2,3,4]))
print(np.polydiv(p1,p2))
#(poly1d([0.]), poly1d([1., 2., 3.]))
(9)polyfit(x,y,k)函数用于多项式拟合,其中x,y分别为要拟合的两组数据,k为拟合多项式中最高次幂
x = np.array([0.0,1.0,2.0,3.0,4.0,5.0])
y = np.array([0.0,0.8,0.9,0.1,-0.8,-1.0])
parray = np.polyfit(x,y,3) #用polyfit返回一个拟合多项式的系数数组
parray
#array([ 0.08703704, -0.81349206, 1.69312169, -0.03968254])
p=np.poly1d(parray) #利用返回的数组生成相应的多项式
print(p)
# 3 2
#0.08704 x - 0.8135 x + 1.693 x - 0.03968
三、数组的文件输入和输出