from matplotlib.collections import PolyCollection

from matplotlib.colors import colorConverter

import matplotlib.pyplot as plt

import numpy as np

np.random.seed(40040157) # Used to allow repeatable experiments (plots)

fig = plt.figure()

ax = fig.gca(projection='3d')

cc = [colorConverter.to_rgba(c,alpha=0.6) for c in ('r','g','b','c','y','m','k')]

ncc = len(cc)

nxs = 5

xs = np.arange(1, nxs+1, 1) # (X's)

# Add boundary X's

xs = np.insert(xs,0,1);

xs = np.append(xs,nxs)

# Create array for Z's

ys = np.zeros(len(xs))

# Create list for all Y's

npolys = 9

zs = [zs+1 for zs in range(npolys)] # Y coordinates (list of integers)

# Create list of colors (cyclic) for all polygons

colorlist = [cc[j%ncc] for j in range(npolys)]

verts = []

# Generate vertices for polygons

for j in zs: # loop on polys

ys = np.random.rand(len(ys))

ys[0], ys[-1] = 0,0 # end points for filled polygons (1,0),(n,0)

verts.append(zip(xs, ys))

poly = PolyCollection(verts, facecolors = colorlist)

poly.set_alpha(0.7)

ax.add_collection3d(poly, zs=zs, zdir='y')

# Right-handed coordinate system

ax.set_xlabel('X') # points to right (X)

ax.set_xlim3d(0, nxs+1)

ax.set_ylabel('Y') # points into screen (Y)

ax.set_ylim3d(0, npolys+1)

ax.set_zlabel('Z') # points up (Z)

ax.set_zlim3d(0, 1)

plt.show()

and this works fine. I then tried to produce a 3D plot of the same form as this one,

segments for each of the 9 polygons). I thought this would be easily accomplished, by replacing PolyCollection with LineCollection. However, I have been unable to produce a 3D plot for line segments connecting the tops of the polygons.

Note, I am using Python 2.6.6, numpy 1.5.1, and matplotlib 1.0.1.

Any help on producing 3D line segment plot would be appreciated.