本文整理汇总了Python中sympy.Matrix.atoms方法的典型用法代码示例。如果您正苦于以下问题：Python Matrix.atoms方法的具体用法？Python Matrix.atoms怎么用？Python Matrix.atoms使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类sympy.Matrix的用法示例。
在下文中一共展示了Matrix.atoms方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: solve_for_sub_subspace
# 需要导入模块: from sympy import Matrix [as 别名]
# 或者: from sympy.Matrix import atoms [as 别名]
def solve_for_sub_subspace(matrixrows, sub_sub_space,
coeffs, cvector,
iofvars, subs_rules,
fvargen, newfvars, tempgen, tempvars):
sspacedict = dict(zip(sub_sub_space, range(len(sub_sub_space))))
length = len(sub_sub_space)
augmatrixrows = []
rownumstore = [] #for debugging
for rownum, row in matrixrows.items():
if row and row[0][0] in sub_sub_space:
augmatrixrows.append(length*[0]+[cvector[rownum]])
rownumstore.append(rownum)
for el in row:
augmatrixrows[-1][sspacedict[el[0]]] = el[1]
fvars = [coeffs[ind] for ind in sub_sub_space]
augmatrix = Matrix(augmatrixrows)
oldfvars = []
if iofvars:
#ipdb.set_trace()
atoms = augmatrix.atoms(sympy.Symbol)
for iofvar in subs_rules:
if iofvar in atoms:
augmatrix = augmatrix.xreplace({iofvar: subs_rules[iofvar]})
atoms = augmatrix.atoms(sympy.Symbol)
for iofvar in iofvars:
if iofvar not in subs_rules and iofvar in atoms:
fvars.append(iofvar)
oldfvars.append(iofvar)
augmatrix = augmatrix.col_insert(-1, sympy.zeros(augmatrix.rows,1))
for row_ind in range(len(augmatrix[:,0])):
coeff_val = -sympy.expand(augmatrix[row_ind,-1]).coeff(iofvar)
augmatrix[row_ind,-2] = coeff_val
augmatrix[row_ind,-1] += coeff_val*iofvar
sols, success = linear_solve(augmatrix, fvars, iofvars, fvargen, newfvars, tempgen, tempvars, len(oldfvars))
if not success:
print(repr(augmatrix))
print(fvars)
print(rownumstore)
print(iofvars)
print(subs_rules)
raise ValueError("Failure. No solutions.")
for oldfvar in oldfvars:
if oldfvar in sols:
subs_rules.update({var: rule.xreplace({oldfvar: sols[oldfvar]})
for var, rule in subs_rules.items()})
subs_rules[oldfvar] = default_simplify(sols[oldfvar])
return sols