在Python 3中绘制隐式方程

Question

我需要在Python3中绘制一些曲线(我很习惯matplotlib.pyplot)但我以前从未画过这些东西,我真的很感激一些提示(特别是提示如何以"整洁"的方式编写代码)和帮助.

有一个例子(让我们用一颗心!):

x^2+(5y/4-sqrt[|x|])^2=1  

我怎么能编码这样的东西？我应该将该公式切割成正常区域然后绘制它们吗？

Answer 1

根据你告诉你要绘制一个隐函数,你应该使用轮廓考虑方程F = x^2和G = 1-(5y/4-sqrt[|x|])^2,然后F-G = 0

import matplotlib.pyplot as plt
import numpy as np

delta = 0.025
xrange = np.arange(-2, 2, delta)
yrange = np.arange(-2, 2, delta)
X, Y = np.meshgrid(xrange,yrange)

# F is one side of the equation, G is the other
F = X**2
G = 1- (5*Y/4 - np.sqrt(np.abs(X)))**2
plt.contour((F - G), [0])
plt.show()

输出: