在 pyplot 中控制等高线标签格式答案

【问题标题】：Controlling contour label formatting in pyplot在 pyplot 中控制等高线标签格式
【发布时间】：2016-10-18 14:31:10
【问题描述】：

我正在绘制一个在[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)

【问题讨论】：

标签： python matplotlib

【解决方案1】：

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

如果您没有花哨的要求，您可以只传递fmt='%d' 而不是自定义类。

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

【讨论】：