ax.legend() 中显示 avxline 竖线

Note

This piece of code was generated by GPT-4.

import matplotlib.pyplot as plt
import matplotlib.lines as mlines
from matplotlib.legend_handler import HandlerLine2D
 
# Step 1: Draw the axvline
fig, ax = plt.subplots()
line = ax.axvline(x=0.5, color='blue', linestyle='--', label='My VLine')
 
# Step 2: Create a custom handler
class HandlerVLine(HandlerLine2D):
    def create_artists(self, legend, orig_handle, xdescent, ydescent, width, height, fontsize, trans):
        # Create a line that matches the axvline style
        line = mlines.Line2D([width / 2, width / 2], [0, height], linestyle=orig_handle.get_linestyle(), 
                             linewidth=orig_handle.get_linewidth(), color=orig_handle.get_color())
        return [line]
 
# Step 3: Assign the handler to the legend
ax.legend(handler_map={mlines.Line2D: HandlerVLine()})
 
plt.show()