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()])
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():
y = np.array(data/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()
def test(danwei):
plt.gca().yaxis.set_major_formatter(mticker.FormatStrFormatter(f'%.1f {danwei}'))
页:
[1]