HeartSurface & matplotlib

来源：互联网发布：数据新闻网崔岸雍编辑：程序博客网时间：2024/06/12 01:18

先上图：

再上代码：

import sysfrom numpy import *import pylab as pfrom mpl_toolkits.mplot3d.axes3d import Axes3Ddef heart(x, y, z): return (x*x + 9/4.*y*y + z*z -1)**3 - x*x*z*z*z - 9/80.*y*y*z*z*z;def polar_to_cartesian(u,v): return (cos(u)*sin(v), sin(u)*sin(v), cos(v))def surf(equation, r, iter=50): us = r_[0:2*pi:iter*2j] vs = r_[0:pi:iter*1j] xs = zeros((len(us),len(vs))) ys = xs.copy() zs = xs.copy() for i in range(len(us)): for j in range(len(vs)): c = polar_to_cartesian(us[i],vs[j]) minrad = min(r_[0:r:iter*2j], key=lambda rad:abs(equation(rad*c[0], rad*c[1], rad*c[2]))) xs[i][j] = minrad*c[0] ys[i][j] = minrad*c[1] zs[i][j] = minrad*c[2] return (xs,ys,zs)heartsurf = surf(heart, 2)fig=p.figure()ax = Axes3D(fig)ax.plot_wireframe(*heartsurf, color='r')#ax.plot3D(*map(ravel, heartsurf))p.s

1. 啥是HeartSurface?

http://mathworld.wolfram.com/HeartSurface.html

2. numpy不能解代数方程，只能暴力求解 -_-!!

3. 画3D图形估计mayavi给力，有机会试用一下