Question

我正在编写一个程序，它有两个输出：一个GUI和一个打印的纸质报告（一个简单的pdf打印出来）。
在两个输出上，我想有一个图表，但有不同的风格。

dark_background（http://matplotlib.org/examples/style_sheets/plot_dark_background.html）
和文章上的五十五夜http://matplotlib.org/examples/style_sheets/plot_fivethirtyeight.html

不知何故，我无法生成具有适当样式的2张图像。其中只有一个是正确的。我没有足够的经验值来发布图片。所以我只发布我的代码。

我的第一个想法是：

import numpy as np
import matplotlib.pyplot as plt

def plot():
        #set pic size
        fig = plt.figure(figsize=(16, 9), dpi=100)
        ax = plt.subplot(111)
        # set x data
        x = range(10)
        # set y data
        y1 = np.zeros(10)
        y2 = [0,1,2,3,4,5,6,7,8,9]
        y3 = [10,9,8,7,6,5,4,3,2,1]
        #plot as errorbar
        ax.errorbar(x, y1, fmt='o', color='green', markersize=8, label='Normal')
        ax.errorbar(x, y2, yerr=0.1, fmt='o', color='orange', markersize=8, label='abw_up')
        ax.errorbar(x, y3, yerr=0.1, fmt='o', color='purple', markersize=8,label='abw_down')
        # limits
        ax.axhline(0.1*10, color='red', lw=2)
        ax.axhline(-0.1*10, color='red', lw=2)
        #set limit of y-Axis
        ax.set_ylim((-1.3,1.3))
        # Labels
        ax.set_xlabel('points number')
        ax.set_ylabel('values')
        # legend
        legend=ax.legend(loc=('upper center'), shadow='true',bbox_to_anchor=(0.5, 1.05),ncol=3, fancybox=True)

        plt.style.use('dark_background')
        plt.savefig('result_dark.png')

        plt.style.use('fivethirtyeight')
        plt.savefig('result_white.png')

但它没有正常工作。其中一张图片是正确的。第二个具有正确的背景颜色，但图例/标签的字体颜色没有改变。我试图分开2张图片，结果是一样的：

import numpy as np
import matplotlib.pyplot as plt
import os

def plot():
    #set pic size

    ax = plt.subplot(111)
    # set x data
    x = range(10)
    # set y data
    y1 = np.zeros(10)
    y2 = [1,2,3,1,2,3,1,2,3,1]
    y3 = [3,1,2,3,1,2,3,1,2,3]
    #plot as errorbar
    ax.errorbar(x, y1, fmt='o', color='green', markersize=8, label='Normal')
    ax.errorbar(x, y2, yerr=0.2, fmt='o', color='orange', markersize=8, label='abw_up')
    ax.errorbar(x, y3, yerr=0.1, fmt='o', color='purple', markersize=8,label='abw_down')
    # limits
    ax.axhline(0.1*10, color='red', lw=2)
    ax.axhline(-0.1*10, color='red', lw=2)
    #set limit of y-Axis
    ax.set_ylim((-1.3,5.3))
    # Labels
    ax.set_xlabel('Messpunkte-Nr.\nMeasurement points number')
    ax.set_ylabel('Spezifikationsgrenze normiert\nnormed to specification')
    # legend
    legend=ax.legend(loc=('upper center'), shadow='true',bbox_to_anchor=(0.5, 1.05),ncol=3, fancybox=True)
    texts =legend.get_texts()
    texts[0].set_color('green')
    texts[1].set_color('orange')
    texts[2].set_color('purple')

fig = plt.figure(figsize=(16, 9), dpi=100)
plt.style.use('dark_background')
plot()
plt.savefig('result_dark.png')
plt.clf()
#plt.close()

fig = plt.figure(figsize=(16, 9), dpi=100)
plt.style.use('fivethirtyeight')
plot()
plt.savefig('result_white.png')
plt.clf()
#plt.close()

我应该如何修复我的代码以使2个图像具有相同的值但风格不同？

Answer 1

我建议你构建类似的代码：

from matplotlib.style import context

def my_plot_function(ax, data, style):
    # do all of your plotting in here, should be fast as no computation
    pass

with context('dark'):
    fig, ax = plt.subplots(1, 1)
    my_plot_function(ax, data, style)
    fig.savefig(...)

with context('fivethirtyeight'):
    fig, ax = plt.subplots(1, 1)
    my_plot_function(ax, data, style)
    fig.savefig(...)

这是一个设计功能，而不是错误。几乎所有由rcparams控制的值都是在对象创建时设置的，而不是绘制时间，因为在渲染时你的图形看起来像是依赖于全局状态是可怕的。这也允许您使用rcparams的上下文管理器，如上所示。调用use只会覆盖样式表显式设置的值（这也是一个设计功能，因此您可以应用多个样式a-la级联样式表）。

Answer 2

所以你的问题似乎是你做了很多密谋，然后告诉pylab你希望你的情节具有特定的风格。该指令似乎没有更新所有内容。所以相反，告诉它你想要使用一种特定的风格。然后情节。然后清除一切。然后重新绘制。

import numpy as np
import matplotlib.pyplot as plt

def plot():
        #set pic size
        fig = plt.figure(figsize=(16, 9), dpi=100)
        ax = plt.subplot(111)
        # set x data
        x = range(10)
        # set y data
        y1 = np.zeros(10)
        y2 = [0,1,2,3,4,5,6,7,8,9]
        y3 = [10,9,8,7,6,5,4,3,2,1]
        #plot as errorbar
        ax.errorbar(x, y1, fmt='o', color='green', markersize=8, label='Normal')
        ax.errorbar(x, y2, yerr=0.1, fmt='o', color='orange', markersize=8, label='abw_up')
        ax.errorbar(x, y3, yerr=0.1, fmt='o', color='purple', markersize=8,label='abw_down')
        # limits
        ax.axhline(0.1*10, color='red', lw=2)
        ax.axhline(-0.1*10, color='red', lw=2)
        #set limit of y-Axis
        ax.set_ylim((-1.3,1.3))
        # Labels
        ax.set_xlabel('points number')
        ax.set_ylabel('values')
        # legend
        legend=ax.legend(loc=('upper center'), shadow='true',bbox_to_anchor=(0.5, 1.05),ncol=3, fancybox=True)


plt.style.use('dark_background')
plot()
plt.savefig('result_dark.png')

plt.clf()
plt.style.use('fivethirtyeight')
plot()
plt.savefig('result_white.png')

这会给你想要的吗？这是我得到的数字。

result_dark

result_white

多个savefig（）具有不同的样式

2 个答案: