最近接觸到一些工學(xué)上的數(shù)學(xué)計算,琢磨著如何用R來輕松秒殺,總結(jié)出了一些方法。
1. 使用逆行列矩陣
舉個例子,解下面的二元一次方程
2x+3y=8
x-2y=5
這個用最簡單的線性代數(shù)就行,Av=b。
R的話用到solve來求逆行列。
> A=matrix(c(2,1,3,-2),2,2)
> A
[,1] [,2]
[1,] 2 3
[2,] 1 -2
> b=c(8,5)
> b
[1] 8 5
> solve(A,b)
[1] 4.4285714 -0.2857143
三元方程的話也是一樣的,只不過A多了一列而已。
2. 用nleqslv函數(shù)
R還有專門的求解方程的包nleqslv。當然不只是解方程,求最優(yōu)值什么的也可以用這個包。安裝完了包來解決同樣的問題。
g <- function(z)
{ x = z[1]
y = z[2]
f1 = 2*x+3*y-8
f2 = x-2*y-5
c(f1,f2)
}
nleqslv(c(1, 1), g) # 制定起始值
可以求得同樣的結(jié)果
> result<-nleqslv(c(1, 1), g)
> result$x
[1] 4.4285714 -0.2857143
當然這只是最簡單的例子,這個包主要用于計算高次元非線形方程。
比方說
3x+xy=15
x*x+2y=20
這個時候解肯定不止一個,用R的話可以先把這兩個函數(shù)可視化一下,然后用可交互界面鼠標點擊兩個函數(shù)的交點,給出初始值,求解。
x=seq(-10, 20, 0.5)
y=seq(-10,20,0.5)
f1=function(x,y) 3*x+x*y - 15
f2=function(x,y) x^2+y*2-20
z1 = outer(x, y, f1)
z2 = outer(x, y, f2)
contour(x, y, z1, levels=0,ylim=c(-10,20))
contour(x, y, z2, levels=0,col=2,add=T)
abline(h=0,col=8); abline(v=0,col=8)
grid()
#定義聯(lián)列方程式
library(nleqslv)
g= function(z) {
x = z[1]
y = z[2]
f1 = 3*x+ x*y - 15 # 方程式1
f2 = x^2+y*2-20 # 方程式2
c(f1, f2)
}
#求解
p=locator(1); points(p,col=4) # 指定初始值
x0=p$x ; y0=p$y # 鼠標點擊指定初始值
ans = nleqslv(c(x0,y0), g) #求解
ans$x
points(ans$x[1],ans$x[2],pch=20,col=2)
q=locator(1); # 用鼠標點擊指定text的位置
text(q$x,q$y,paste("x=",round(ans$x[1],3),"y=",round(ans$x[2],3)))

以上