前面用 C 写了个库, 矩阵的基本运算都包涵了, +-*/ 求逆, rref(行最简形)等, 并且利用这个库实现了另外一些功能, 如 线性方程组求解, 线性回归, 线性规划(正在写, 个人感觉比较难)
可是使用的时候及其不便, 由于文本解析问题。 最近学了些python, 于是发现这个用来解析文本不错, 于是就写了个。
只要达到系数变量分离就好办了。
#!/usr/bin/env python
#coding=utf-8
import re
'''
coeffTable 是 {变量:系数列表}, 如果不懂, 请运行看程序结果就懂了
建议看之前先运行一下看看结果
'''
coeffTable = {'__LENGTH__': 0, '__CONS__':[]}
def parseEquation(str):
# 分离系数与变量
pattern = r'(^\s*\d*|[+-]\s*\d*)\s*(\w*)' # 用来分离变量与系数的表达式。
ptrn = re.compile(pattern)
left, right = str.split('=') # 方程左右两边分离
coeffTable['__CONS__'].append(float(right.strip())) # 右边加入到常量
# 请将 1, 2, 3 结合起来看, 比较容易懂
# 1. 原有变量的系数列表在后面添加一个0, 同时列表长度增加 1
for key in coeffTable:
if key not in ('__LENGTH__', '__CONS__'):
coeffTable[key].append(0)
coeffTable['__LENGTH__'] += 1
# 遍历在新方程中出现的变量并转换其对应系数
for val in ptrn.findall(left):
# 转换系数
if val[0] == '-': coeff = 1
elif val[0][-1].isspace():
if val[0][0] == '-': coeff = -1
else: coeff = 1
else:
coeff = float(''.join(val[0].split())) # 去掉所有空白字符后转换成 float 数
varName = ''.join(val[1].split()) #去掉所有空白字符,获得变量名
coffList = coeffTable.get(varName)
# 2. 将新出现的变量添加至系数表
if coffList == None:
coeffTable[varName] = []
for i in range(coeffTable.get('__LENGTH__')):
coeffTable[varName].append(0)
# 3. 将该变量的系数列表的最后一个系数变成当前方程对应的系数
coeffTable[varName][-1] = coeff
# 测试
if __name__ == '__main__':
equation = '3x + 4y - z - a + b + 3c = 7'
print(equation)
parseEquation(equation)
equation = '7x - 6y = 9'
print(equation)
parseEquation(equation)
print(coeffTable)