控制pyplot中的轮廓标签格式

时间:2016-10-18 02:24:37

标签: python matplotlib

我正在制作一个情节,我在[2000, 4000, 6000, 8000]处有轮廓。轮廓标记为2000.000,4000.000等。我想摆脱所有那些尾随的零。我现在可以找到的最佳选项是http://matplotlib.org/examples/pylab_examples/contour_label_demo.html,它建议为标签定义一个新类,控制它们的显示方式,然后使用该类。我之前从未在python中看到过如此复杂的选项。有没有更直接的方式?

以下是为标签定义类的示例。

import matplotlib
import numpy as np
import matplotlib.cm as cm
import matplotlib.mlab as mlab
import matplotlib.ticker as ticker
import matplotlib.pyplot as plt

matplotlib.rcParams['xtick.direction'] = 'out'
matplotlib.rcParams['ytick.direction'] = 'out'

##################################################
# Define our surface
##################################################
delta = 0.025
x = np.arange(-3.0, 3.0, delta)
y = np.arange(-2.0, 2.0, delta)
X, Y = np.meshgrid(x, y)
Z1 = mlab.bivariate_normal(X, Y, 1.0, 1.0, 0.0, 0.0)
Z2 = mlab.bivariate_normal(X, Y, 1.5, 0.5, 1, 1)
# difference of Gaussians
Z = 10.0 * (Z2 - Z1)

##################################################
# Make contour labels using creative float classes
# Follows suggestion of Manuel Metz
##################################################
plt.figure()

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


# Define a class that forces representation of float to look a certain way
# This remove trailing zero so '1.0' becomes '1'
class nf(float):
    def __repr__(self):
        str = '%.1f' % (self.__float__(),)
        if str[-1] == '0':
            return '%.0f' % self.__float__()
        else:
            return '%.1f' % self.__float__()

# Recast levels to new class
CS.levels = [nf(val) for val in CS.levels]

# Label levels with specially formatted floats
if plt.rcParams["text.usetex"]:
    fmt = r'%r \%%'
else:
    fmt = '%r %%'
plt.clabel(CS, CS.levels, inline=True, fmt=fmt, fontsize=10)

1 个答案:

答案 0 :(得分:1)

fmt参数可以是经典格式字符串,也可以是将标量转换为字符串的可调用字符。

如果您没有花哨的要求,可以通过fmt='%d'而不是自定义类。

对于常见格式,您还应该在matplotlib.ticker之前使用默认格式化程序,然后才能实现自己的格式。