This example extends the
Surface Chart (1) example by adding surface grid and contour lines, and using alternative view angles and continuous coloring.
pythondemo\surface2.py
#!/usr/bin/python
# The ChartDirector for Python module is assumed to be in "../lib"
import sys, os
sys.path.insert(0, os.path.join(os.path.abspath(sys.path[0]), "..", "lib"))
from pychartdir import *
import math
# The x and y coordinates of the grid
dataX = [0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0]
dataY = [0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0]
# The values at the grid points. In this example, we will compute the values using the formula z =
# sin((x - 0.5) * 2 * pi) * sin((y - 0.5) * 2 * pi)
dataZ = [0] * (len(dataX) * len(dataY))
for yIndex in range(0, len(dataY)) :
y = (dataY[yIndex] - 0.5) * 2 * 3.1416
for xIndex in range(0, len(dataX)) :
x = (dataX[xIndex] - 0.5) * 2 * 3.1416
dataZ[yIndex * len(dataX) + xIndex] = math.sin(x) * math.sin(y)
# Create a SurfaceChart object of size 720 x 540 pixels
c = SurfaceChart(720, 540)
# Add a title to the chart using 20 points Times New Roman Italic font
c.addTitle("Quantum Wave Function", "Times New Roman Italic", 20)
# Set the center of the plot region at (360, 245), and set width x depth x height to 360 x 360 x 270
# pixels
c.setPlotRegion(360, 245, 360, 360, 270)
# Set the elevation and rotation angles to 20 and 30 degrees
c.setViewAngle(20, 30)
# Set the data to use to plot the chart
c.setData(dataX, dataY, dataZ)
# Spline interpolate data to a 80 x 80 grid for a smooth surface
c.setInterpolation(80, 80)
# Set surface grid lines to semi-transparent black (dd000000)
c.setSurfaceAxisGrid(0xdd000000)
# Set contour lines to semi-transparent white (80ffffff)
c.setContourColor(0x80ffffff)
# Add a color axis (the legend) in which the left center is anchored at (645, 270). Set the length
# to 200 pixels and the labels on the right side. Use smooth gradient coloring.
c.setColorAxis(645, 270, Left, 200, Right).setColorGradient()
# Set the x, y and z axis titles using 10 points Arial Bold font
c.xAxis().setTitle("x/L(x)", "Arial Bold", 10)
c.yAxis().setTitle("y/L(y)", "Arial Bold", 10)
c.zAxis().setTitle("Wave Function Amplitude", "Arial Bold", 10)
# Output the chart
c.makeChart("surface2.jpg")
© 2021 Advanced Software Engineering Limited. All rights reserved.