Маркировка уровней контуров с помощью персонализированных меток

Question

Я бы хотел обозначить уровни контуров на графике плотности персонализированными метками.Я написал что-то вроде

contours = plt.contour(X, Y, Z, 3, colors='black')
plt.clabel(contours, inline=True, levels=[1,2], fontsize=8)

plt.imshow(Z, extent=[0, 5, 0, 5], origin='lower',
       cmap='RdGy', alpha=0.5)
plt.colorbar();

Мне бы хотелось, чтобы контуры были помечены именем, например r'\textbf{1-}'+r'$\sigma$'

Answer 1

То, что вы запрашиваете, непосредственно показано во втором примере contour_label_demo .Я просто приведу этот пример здесь:

##################################################
# Label contours with arbitrary strings using a
# dictionary
##################################################
plt.figure()

# Basic contour plot
CS = plt.contour(X, Y, Z)

fmt = {}
strs = ['first', 'second', 'third', 'fourth', 'fifth', 'sixth', 'seventh']
for l, s in zip(CS.levels, strs):
    fmt[l] = s

# Label every other level using strings
plt.clabel(CS, CS.levels[::2], inline=True, fmt=fmt, fontsize=10)