How to plot points on top of cplot?

[kaceyaurum]

I used cplot to graph a complex function, and I want to mark certain points in the complex plane. How can I do this?

[nschloe]

How would you plot the points with mpl?

[kaceyaurum]

I would like to use matplotlib.pyplot.scatter, but I am not sure how to format my input with the complex plane.

[nschloe]

mpl doesn't know it's a complex plane. You can just scatterplot over the cplot and x-y-coordinates.

[kaceyaurum]

I want to do something like:
ax = cplot.plot(f, (-2.0, +2.0, 400), (-2.0, +2.0, 400))
ax.scatter([.5], [0], s=100, marker='.', c='red')
plt.show(ax)

but this puts the points in the arg scale bar. How can I put the points on the main plot?

[nschloe]

Ah, I see. cplot.plot doesn't return the ax, but plt. You'll have to manually find the correct axis to plot to. Maybe it should return ax? Not sure.

[onnoeberhard]

It's possible to find the original axis by ax = plt.gcf().get_axes()[0]. But I do agree that the function should return the ax object (or even accept an ax argument) rather than the plt module.