python matplotlib画双阱函数图

这是一个关于 Python Matplotlib 画双阱函数图的问题，可以回答。以下是一个示例代码：

import numpy as np
import matplotlib.pyplot as plt

def double_well_potential(x, a, b):
    return a * x ** 4 - b * x ** 2

x = np.linspace(-2, 2, num=100)
y = double_well_potential(x, a=1, b=1)

plt.plot(x, y)
plt.xlabel('x')
plt.ylabel('Potential Energy')
plt.title('Double Well Potential Energy Function')
plt.show()

这个函数用于画一个具有双阱的势能函数图，可用于展示分子动力学模拟中的分子的势能面。