rcparams 未应用于自定义 matplotlib 类

问题描述 投票:0回答:1

我正在尝试围绕

matplotlib.figure.Figure
编写一个自定义图形类,其中自动应用正确的格式。这是当前配置:

import matplotlib
from matplotlib.axes import Axes
from matplotlib.figure import Figure
from matplotlib.backends.backend_qtagg import FigureCanvasQTAgg as Canvas

class CustomFigure(Figure):
    def __init__(self, figsize: tuple, layout: str):
        super().__init__(figsize=figsize, layout=layout)
        self.canvas = Canvas(self)
        matplotlib.use("QtAgg")
        self.set_common_params()
        
        
    def generate_axes(self, num_axes: int, layout: tuple = None) -> Axes:
        if layout is None:
            layout = (1, num_axes)
        return self.subplots(*layout)
    
        
    def set_common_params(self):
        matplotlib.rcParams["figure.titlesize"] = 45
        matplotlib.rcParams["axes.titlesize"]   = 13
        matplotlib.rcParams["axes.labelsize"]   = 12
        matplotlib.rcParams["axes.linewidth"]   = 1.5
        matplotlib.rcParams["xtick.labelsize"]  = 11
        matplotlib.rcParams["ytick.labelsize"]  = 11
        
    
    @staticmethod
    def set_labels(ax: Axes, xlabel: str, ylabel: str, title: str = None):
        ax.set_xlabel(xlabel)
        ax.set_ylabel(ylabel)
        if title is not None:
            ax.set_title(title)
            
    
    def generate_pdf(self, filename: str):
        self.savefig(f"{filename}.pdf")
        
        


if __name__ == "__main__":
    import sys
    from PySide6.QtWidgets import QApplication, QMainWindow
    app = QApplication(sys.argv)
    win = QMainWindow()
    fig = CustomFigure((5,5), "tight")
    fig.set_labels(fig.generate_axes(1), "X", "Y", "Title")
    win.setCentralWidget(fig.canvas)
    win.show()
    sys.exit(app.exec())

我尝试将 rcParams 放在代码的每个可能的位置,甚至在类定义之前,但没有任何效果。

如何正确应用 rcParams?

python matplotlib
1个回答
0
投票

如果我正确理解你的问题,你的代码确实根据需要设置了除了图标题之外的参数。那是因为您正在使用具有“副标题”的子图。所以你为

figure.titlesize
设置的参数将不起作用!检查这个:

import matplotlib
from matplotlib.axes import Axes
from matplotlib.figure import Figure
from matplotlib.backends.backend_qtagg import FigureCanvasQTAgg as Canvas
import matplotlib.pyplot as plt


class CustomFigure(Figure):
    def __init__(self, figsize: tuple, layout: str):
        super().__init__(figsize=figsize, layout=layout)
        self.canvas = Canvas(self)
        matplotlib.use("QtAgg")
        self.set_common_params()

    def generate_axes(self, num_axes: int, layout: tuple = None) -> Axes:
        if layout is None:
            layout = (1, num_axes)
        return self.subplots(*layout)

    def set_common_params(self):
        matplotlib.rcParams["figure.titlesize"] = 'large'  # this will not work since it is not for a subplot
        matplotlib.rcParams["axes.titlesize"] = 45
        matplotlib.rcParams["axes.labelsize"] = 45
        matplotlib.rcParams["axes.linewidth"] = 1.5
        matplotlib.rcParams["xtick.labelsize"] = 10
        matplotlib.rcParams["ytick.labelsize"] = 10
        matplotlib.rcParams["font.sans-serif"] = "Verdana"

    @staticmethod
    def set_labels(ax: Axes, xlabel: str, ylabel: str, title: str = None):
        ax.set_xlabel(xlabel)
        ax.set_ylabel(ylabel)
        if title is not None:
            ax.set_title(title)

    def generate_pdf(self, filename: str):
        self.savefig(filename)

    def clearfig(self):
        self.clf()


if __name__ == "__main__":
    import sys
    from PySide6.QtWidgets import QApplication, QMainWindow

    app = QApplication(sys.argv)
    win = QMainWindow()
    fig = CustomFigure((5, 5), "tight")
    fig.set_labels(fig.generate_axes(1), "X", "Y", "Title")
    win.setCentralWidget(fig.canvas)
    win.show()
    sys.exit(app.exec())
© www.soinside.com 2019 - 2024. All rights reserved.