使用 matplotlib 将语法样式文本添加到 Python 中的 plot

Question

I'm wanting to incorporate a text box with code style highlight in a plot in matplotlib, and I'm wondering whether there's a method to do so? Thank you in advance for your consideration and suggestions.

Answer 1

Yes, I can try like this:

import matplotlib.pyplot as plt

#create data
x = [3, 6, 8, 12, 14]
y = [4, 9, 14, 12, 9]

#create scatterplot
plt.scatter(x, y)

font = {'family': 'serif',
        'color':  'red',
        'weight': 'bold',
        'size': 10
        }

box = {'facecolor': 'none',
       'edgecolor': 'green',
       'boxstyle': 'round'
      }

#add text with custom font

code = """
import numpy as np 
import pandas as pd

"""
#add text with custom font
plt.text(6, 9.5, code, fontdict=font, bbox=box)

Output look like:

Reference: https://www.statology.org/matplotlib-add-text/