python绘图函数返回_在Python中使用pyplot返回复杂值的绘图函数

我在Facebook上发现了这个图像,想要绘制它,因为我认为这将是一个学习新东西的好机会.

代码：

import numpy as np

import scipy.special

import pylab

x = np.linspace( -1.0, 1.0, 200 )

y = np.sqrt( 1-np.square(np.complex64(x) ) ) + np.complex64( scipy.special.cbrt( np.square( x ) ) )

pylab.plot( np.complex64( x ), y )

pylab.show()

输出：

IMG http://i45.tinypic.com/33ab5g2.png

我无法获得图中给出的情节.为了绘图,我将公式重新排序为y = f(x)形式并将其转换为上面的代码.

import matplotlib.pyplot as plt

import numpy as np

y, x = np.ogrid[-1:2:100j, -1:1:100j]

plt.contour(x.ravel(), y.ravel(), x**2 + (y-((x**2)**(1.0/3)))**2, [1])

plt.axis('equal')

plt.show()

您可以使用宽高比来使曲线更像心脏：

fig = plt.figure()

ax = fig.add_subplot(1, 1, 1)

y, x = np.ogrid[-1.5:2:1000j, -2:2:1000j]

plt.contour(x.ravel(), y.ravel(), x**2 + (y-((x**2)**(1.0/3)))**2, [1])

ax.set_aspect(0.75)

plt.show()