Matplotlib features - bonus

October 10, 2024

I cannot be talking about LaTeX font without giving an example on how to achieve it. On top of that, I added a few changes to make the plot look more to what can be obtained with ROOT (without having to deal with its functions). Here is the result:

figure with root features

And here is the code:

     
import matplotlib.pyplot as plt
from matplotlib.ticker import MultipleLocator

# set the LaTeX font
plt.rcParams.update({
    'text.usetex': True,
    'font.family': 'Helvetica'
})

# create two sets of data 
plot_x = [0, 1, 2, 3, 4, 
          5, 6, 7, 8]
plot_y = [400, 100, 400, 300, 400, 
          500, 400, 700, 400]

# create a figure 
fig, ax=plt.subplots(figsize=(8, 6))

# plot data 
plt.plot(plot_x, plot_y, linestyle='--', 
         marker='s', color='#224afb', 
         label='Curve 1')

# add a grid 
ax.yaxis.set_minor_locator(MultipleLocator(20))
ax.grid(which='major', axis='both', 
              linestyle=':', color='black')

# set font sizes 
font_size = 20
ax.yaxis.get_offset_text().set_fontsize(font_size)
plt.title('Another and last suggestion', 
          fontsize=font_size)
plt.xlabel('X-axis', fontsize=font_size, 
           horizontalalignment='right', x=1.0)
plt.ylabel('Y-axis', fontsize=font_size, 
           horizontalalignment='right', y=1.0)
plt.xticks(fontsize=font_size)
plt.yticks(fontsize=font_size)

# add a legend 
plt.legend(loc='lower right', fontsize=font_size, 
           handlelength=1, framealpha=1, 
           edgecolor='black', fancybox=False)

# show the plot 
plt.show()

Bonus point if you can tell me what number is the curve's color.