[已解决]matplotlib y轴刻度+单位

拽猫的尾巴

我想要实现以下效果


用上以下代码后可以实现

import matplotlib.pyplot as plt
import matplotlib.ticker as mticker

plt.gca().yaxis.set_major_formatter(mticker.FormatStrFormatter('%.1f 万'))

但是如果我把单位赋值给一个变量，要怎么操作才能实现上图的效果？（我想要把代码放进一个函数中，单位可以通过改变参数进行改变）
源代码如下：

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import matplotlib.ticker as mticker
from pylab import *
from scipy.interpolate import make_interp_spline as spl

data = pd.read_csv(r'移动行政区总体人口数.csv', encoding = 'gbk')
data = data.iloc[:, 0:-1]

x = np.array([i for i in range(data.shape[0])])
x_smooth = np.linspace(x.min(), x.max(), 300)

#设置图形格式
plt.figure(figsize = (16, 9), dpi = 300)  #改图片尺寸和分辨率

#设置边框格式
ax = plt.gca() #获取边框
ax.spines['top'].set_color('none') #设置上边框为 无色
ax.spines['right'].set_color('none') #设置上边框为 无色

#以固定间隔显示刻度
x_range = np.array(data.iloc[:,0])
_ = plt.xticks(x, x_range)

#设置中文字体、字体大小
plt.rcParams['font.sans-serif'] = ['SimHei'] #指定默认字体 
plt.rcParams['axes.unicode_minus'] = False #解决保存图像是负号'-'显示为方块的问题 
plt.xticks(ticks = a, fontsize = 16) #显示所有x轴的刻度、刻度字体大小
plt.xticks(rotation = 90) #x轴的刻度旋转90度
plt.yticks(fontsize = 16) #显示所有y轴的刻度字体大小
plt.gca().yaxis.set_major_formatter(mticker.FormatStrFormatter('%.0f 万')) #y周刻度后加上中文单位

for i in data.columns.tolist()[1:]:
    y = np.array(data[i]/10000)
    y_smooth = spl(x,y)(x_smooth)
    
    plt.plot(x_smooth, y_smooth, label = i)
    
plt.legend(loc=2, bbox_to_anchor=(1.0,1.0),borderaxespad = 0., prop = {'size':16})  #显示图例、设置图例大小、把图例放到边界外

plt.show()

最佳答案

月排行榜 / 总排行榜

isdkz

2022-4-6 13:03:32

def test(danwei):
    plt.gca().yaxis.set_major_formatter(mticker.FormatStrFormatter(f'%.1f {danwei}'))

跳转到最佳答案楼层

isdkz

def test(danwei):
    plt.gca().yaxis.set_major_formatter(mticker.FormatStrFormatter(f'%.1f {danwei}'))