提交: a504100084ef6dda386173ca6623e7572885f224 - myprojects-hg-reborn (hg)

Commit MetaInfo

• 修訂: a504100084ef6dda386173ca6623e7572885f224 (tree)
• 時間: 2014-01-05 07:32:30
• 作者: Lorenzo Isella <lorenzo.isella@gmai...>
• Commiter: Lorenzo Isella

Log Message

A code for visualizing the spherical harmonics with MayaVi2.

Change Summary

• add: Python-codes/spherical_harmonics.py

差異

diff -r 7cbd25cc7b99 -r a504100084ef Python-codes/spherical_harmonics.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Python-codes/spherical_harmonics.py	Sat Jan 04 23:32:30 2014 +0100
@@ -0,0 +1,52 @@
+"""
+Plot spherical harmonics on the surface of the sphere, as well as a 3D
+polar plot.
+
+This example requires scipy.
+
+In this example we use the mlab's mesh function:
+:func:`mayavi.mlab.mesh`.
+For plotting surfaces this is a very versatile function. The surfaces can
+be defined as functions of a 2D grid.
+
+For each spherical harmonic, we plot its value on the surface of a
+sphere, and then in polar. The polar plot is simply obtained by varying
+the radius of the previous sphere.
+"""
+
+# Author: Gael Varoquaux <[e-mail]>
+# Copyright (c) 2008, Enthought, Inc.
+# License: BSD Style.
+
+from mayavi import mlab
+import numpy as np
+from scipy.special import sph_harm
+
+# Create a sphere
+r = 0.3
+pi = np.pi
+cos = np.cos
+sin = np.sin
+phi, theta = np.mgrid[0:pi:101j, 0:2*pi:101j]
+
+x = r*sin(phi)*cos(theta)
+y = r*sin(phi)*sin(theta)
+z = r*cos(phi)
+
+mlab.figure(1, bgcolor=(1, 1, 1), fgcolor=(0, 0, 0), size=(400, 300))
+mlab.clf()
+# Represent spherical harmonics on the surface of the sphere
+for n in range(1, 6):
+    for m in range(n):
+        s = sph_harm(m, n, theta, phi).real
+
+        mlab.mesh(x-m, y-n, z, scalars=s, colormap='jet')
+
+        s[s<0] *= 0.97
+
+        s /= s.max()
+        mlab.mesh(s*x-m, s*y-n, s*z+1.3, scalars=s, colormap='Spectral')
+
+mlab.view(90, 70, 6.2, (-1.3, -2.9, 0.25))
+mlab.show()
+