Cleanup parasite_simple example.

anntzer · anntzer · commit 935976cf7824 · 2022-08-07T11:30:44.000+02:00
- Use `legend(labelcolor=...)` instead of setting colors manually.
- Use `ax.legend` instead of `plt.legend`.
- Order imports.
diff --git a/examples/axes_grid1/parasite_simple.py b/examples/axes_grid1/parasite_simple.py
@@ -2,13 +2,12 @@
 ===============
 Parasite Simple
 ===============
-
 """
-from mpl_toolkits.axes_grid1 import host_subplot
+
 import matplotlib.pyplot as plt
+from mpl_toolkits.axes_grid1 import host_subplot
 
 host = host_subplot(111)
-
 par = host.twinx()
 
 host.set_xlabel("Distance")
@@ -18,12 +17,9 @@
 p1, = host.plot([0, 1, 2], [0, 1, 2], label="Density")
 p2, = par.plot([0, 1, 2], [0, 3, 2], label="Temperature")
 
-leg = plt.legend()
+host.legend(labelcolor="linecolor")
 
 host.yaxis.get_label().set_color(p1.get_color())
-leg.texts[0].set_color(p1.get_color())
-
 par.yaxis.get_label().set_color(p2.get_color())
-leg.texts[1].set_color(p2.get_color())
 
 plt.show()
