pip install fenics pyfem

from fenics import *
import matplotlib.pyplot as plt
# 創(chuàng)建一個(gè)網(wǎng)格
mesh = UnitSquareMesh(8, 8)
# 定義邊界條件
def boundary(x, on_boundary):
    return on_boundary
bc = DirichletBC(mesh, Constant(0), boundary)
# 定義變量和函數(shù)空間
V = FunctionSpace(mesh, 'P', 1)
u = TrialFunction(V)
v = TestFunction(V)
# 定義方程
f = Constant(-6.0)
a = dot(grad(u), grad(v))*dx
L = f*v*dx
# 求解方程
u = Function(V)
solve(a == L, u, bc)
# 繪制結(jié)果
plot(u)
plt.show()

python復(fù)制代碼from pyfem import Problem
from pyfem.utils.meshGenerator import createRectangleMesh
from pyfem.utils.output import printMatrix
from pyfem.materials.linearElasticity import LinearElasticity
from pyfem.boundaryConditions.dirichlet import Dirichlet
from pyfem.solvers.linearSolver import solveLinearSystem
# 創(chuàng)建問(wèn)題實(shí)例
problem = Problem()
# 創(chuàng)建網(wǎng)格
mesh = createRectangleMesh([0, 1], [0, 1], [0, 1], [10, 10])
problem.setMesh(mesh)
# 定義材料屬性
elasticity = LinearElasticity(E=1, nu=0.3)
problem.setMaterialProperties(elasticity)
# 定義邊界條件
dirichlet = Dirichlet(u=0)
problem.addBoundaryCondition(dirichlet, 'left')
problem.addBoundaryCondition(dirichlet, 'right')
problem.addBoundaryCondition(dirichlet, 'bottom')
problem.addBoundaryCondition(dirichlet, 'top')
# 構(gòu)建系統(tǒng)矩陣和右側(cè)向量
A, b = problem.buildSystem()
# 求解線性系統(tǒng)
u = solveLinearSystem(A, b)
# 打印結(jié)果
printMatrix(A)
printMatrix(b)
printMatrix(u)