matplotlib 颜色条图例标签中的字体大小不一致

g6ll5ycj  于 2023-06-23  发布在  其他
关注(0)|答案(1)|浏览(159)

我正在使用Python中的matplotlib进行数据可视化。我的图中的一个可视元素是一个颜色条图例,它用相应的颜色表示不同的值。但是,我注意到colorbar图例标签的字体大小不一致,妨碍了可读性,影响了情节的整体美感。
为了解决这个问题,我尝试使用colorbar对象的set_label和set_tick_params方法显式地设置字体大小。但是,尽管指定了所需的字体大小,图例标签的字体大小仍然不一致。
有什么建议吗?
下面是我使用的代码-输入是一个字典,其中包含来自图的节点ID和相应的值

import pathlib
from typing import Union, Dict, Optional

import osmnx as ox

from matplotlib import pyplot as plt, colors, cm
from matplotlib.font_manager import FontProperties
from matplotlib.ticker import FuncFormatter

from configs.default import get_default_config
from preprocessing.graph_loader import GraphLoader

root_dir = pathlib.Path(__file__).parent.parent.absolute()

class GraphMapper:
    def __init__(self):
        self.config = get_default_config()['ds']
        self.graph = GraphLoader().graph

    def plot_dict_on_graph_as_heatmap(
            self,
            dictionary: Dict,
            title: str,
            filename: Optional[str] = None,
            font: str = 'Consolas',  # 'Times New Roman',
            fontsize: int = 25
    ) -> None:
        cmap = plt.cm.inferno
        maximum_value = max([dictionary[k] for k in dictionary.keys()])
        minimum_value = max(min(dictionary.values()), 1)
        norm = colors.LogNorm(vmin=minimum_value, vmax=maximum_value)
        sm = cm.ScalarMappable(norm=norm, cmap=cmap)

        node_color = [
            sm.to_rgba(dictionary[n])
            if dictionary[n] > 0
            else sm.to_rgba(minimum_value)
            for n in self.graph.nodes(data=False)
        ]

        node_size = [10 if dictionary[n] > 0 else 0 for n in self.graph.nodes(data=False)]

        # noinspection PyTypeChecker
        fig, ax = ox.plot_graph(
            self.graph,
            node_size=node_size,
            edge_linewidth=.5,
            node_color=node_color,
            edge_color='#333333',
            bgcolor='w',
            show=False,
            close=False,
        )

        cbar = plt.colorbar(sm, ax=ax, shrink=0.8, pad=0.01)
        font_properties = FontProperties(
            family=font,
            size=fontsize,
        )

        if title != '':
            cbar.set_label(title, fontproperties=font_properties)

        # formatter = FuncFormatter(lambda x, _: r"${:.0f} \times 10^3$".format(x / 1e3))
        # cbar.ax.yaxis.set_major_formatter(formatter)
        # cbar.ax.tick_params(labelsize=fontsize)
        # cbar.ax.yaxis.set_tick_params(pad=fontsize)
        cbar.ax.set_yticklabels(cbar.ax.get_yticklabels(), fontproperties=font_properties)

        plt.subplots_adjust(
            left=0.01,
            bottom=0.01,
            right=0.99,
            top=0.99,
            wspace=0.01,
            hspace=0.01
        )

        if filename is not None:
            plt.savefig(root_dir / "visualizations" / (str(filename) + ".pdf"), bbox_inches='tight')

        plt.show()

下面是输出图像:
The value 10^3 has the format I am looking for, while the other values are not correctly formatted.
我尽力了

  • 设置字体属性,使用FontProperties
  • 设置colorbar中每个记号标签的字体属性
for label in cbar.ax.yaxis.get_ticklabels():
    label.set_fontproperties(fontprops)

cbar.set_ticklabels([label.get_text() for label in cbar.ax.get_yticklabels()], fontproperties=fontprops)
  • 我尝试了格式化程序
formatter = FuncFormatter(lambda x, pos: f'{x:.0f}')
cbar.ax.yaxis.set_major_formatter(formatter)
cbar.ax.set_yticklabels(cbar.ax.get_yticklabels(), fontproperties=fontprops)
jecbmhm3

jecbmhm31#

我真的找不出问题出在哪里(特别是其他字典工作得很好),但设置默认字体/字体大小工作得很好:

mpl.rcParams['font.family'] = 'Consolas'
mpl.rcParams['font.weight'] = 'light' 
mpl.rcParams['font.size'] = 25

相关问题