Re: Stata/Python integration part 5: Three-dimensional surface plots of marginal predictions

I am a novice using Python with Stata -- but, in case it helps, I found success by inserting:

from mpl_toolkits.mplot3d import Axes3D
from mpl_toolkits.mplot3d.art3d import Poly3DCollection

before

import matplotlib.pyplot as plt

I hope this helps.