主页 / coding / 问题 / 79032221
Accepted
ohshitgorillas
Asked: 2024-09-28 00:29:25 +0800 CST

rcparams 未应用于自定义 matplotlib 类

  • 772

我正在尝试编写一个自定义图形类，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

1 个回答

  1. Best Answer

    如果我理解正确的话，你的代码确实根据需要设置了参数，除了图标题之外。那是因为你正在使用具有“suptitle”的子图。所以你设置的参数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())
    • 1

相关问题