Somewhere you can use c="k" for colour but sometimes you cannot, you must use color="k".
Some settings only exposed to figure class but not the ax class. And when you are doing stuff towards ax class, you must write
ax.set_ylim(0,3)
instead of
ax.ylim(0,3)
Matplotlib.pyplot is known for its nonsensical api.