|
10鱼币
我想要实现以下效果
用上以下代码后可以实现
- 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()
复制代码
- def test(danwei):
- plt.gca().yaxis.set_major_formatter(mticker.FormatStrFormatter(f'%.1f {danwei}'))
复制代码
|
|