今天传热学老师说到高斯-赛德尔迭代法,我就想拿python写一个小程序来计算。
在网上找例子,发现居然没有人拿python写,都是C / C++ / Matlab的。没有参考答案,真是写得我焦头烂额啊。
截图:
#!/usr/bin/env python
# -*- coding=gb2312 -*-
from __future__ import division #改变除法
import copy
data = open('data.txt')
content = data.read()
a = [map(int,ln.strip().split(' '))
for ln in content.splitlines() if ln.strip()] #形成矩阵二维数组
n = len(a[0])-1 #方程组解的个数
def gui0(temp): #定义归零函数
"此函数的作用是将一个数组中的所有数值清空为0"
for i in range(0,len(temp)):
temp[i] = 0
return temp
def dp(): #定义检验函数:
for i in range(0,n-1):
aabs = map(abs, a[i])
if 2*abs(a[i][i]) + abs(a[i][n]) >= reduce(lambda x, y: x+y, aabs):
del aabs
for j in range(0,n):
if a[i][j] != 0:
pass
else:
print "\n您的数组含有为0系数!\n"
exit()
else:
print "\n您的数组无法满足对角占优。\n"
exit()
print "\n您的数组可以满足对角占优。\n"
t = copy.copy(a[0]) #浅复制
t = gui0(t)
t[0:1] = [] #由于a[0]数组元素数比解数多一个,因此在归零后消去第一个0
def main(): #定义迭代函数
tmp1 = []
for i in range(0,n):
for j in range (0,n):
rest = a[i][j]*t[j]
tmp1.append(rest)
t[i] = (a[i][n] - reduce(lambda x, y: x+y, tmp1) + a[i][i]*t[i]) / a[i][i]
tmp1 = []
print t
if len(raw_input("")) == 0:
pass
else:
exit("用户手动退出。")
print '''
“高斯-赛德尔迭代法” 简易计算程序:
Designed by 安
使用说明:
1.将您的方程组矩阵按照格式写入“data.txt”中,格式举例:
如果您有一个二元一次方程组ax+by=c, dx+ey=f,则data.txt内容为:
a b c
d e f
以此类推,数字与数字之间使用半角空格隔开,行末不需要加空格;
2.程序会自动验证方程组,如果含有为0系数或者无法满足对角占优,程序会自动退出;
3.初解会被设定为0,0,0,0,……;
4.程序每迭代一次之后就会暂停,请按回车继续,直到您取得想要的精度为止,您可以输入任何字符再按回车便可退出程序。
'''
if len(raw_input("按下回车开始计算:")) == 0:
dp()
while True:
main()
else:
exit("用户取消程序运行。")
print u"\n\n结果已经输出!"