【数据分析&数据挖掘】2000-2017年各个产业、行业的柱状图

 1 import matplotlib.pyplot as plt
 2 import numpy as np
 3 
 4 def draw_data(n, a, b, c, d, values, fig, plt, t, xl, ticks):
 5     fig.add_subplot(2, 2, n)
 6     x = np.arange(1, c, 1)
 7     y = values[d, a:b+1]
 8     print("---y---:
",y)
 9     plt.bar(x, y, width=0.5, color=["#BCEE68", "#A2CD5A", "#6E8B3D"])
10     plt.title(t)
11     plt.xlabel(xl)
12     plt.ylabel("生产总值（亿元）")
13     plt.xticks(x, ticks)
14     for i,j in zip(x, y):
15         plt.text(i, j, "%.2f亿元"%j, horizontalalignment='center', verticalalignment='bottom')
16     return plt
17 
18 def show_data():
19     fig = plt.figure(figsize=(20, 12), dpi=120)
20     plt.rcParams['font.sans-serif'] = 'SimHei'
21     plt.rcParams['axes.unicode_minus'] = False
22     # plt.subplots_adjust(hspace=0.3)
23     return fig, plt
24 
25 def save_data():
26     plt.savefig("./2000-2017年各个产业、行业的柱状图.png")
27     plt.show()
28     return None
29 
30 
31 def build_data():
32     res = np.load("./国民经济核算季度数据.npz", allow_pickle=True)
33     columns = res["columns"]
34     values = res["values"]
35     # print(columns)
36     # print(values)
37     return columns, values
38 
39 def main():
40     columns, values = build_data()
41     fig, plt = show_data()
42     title1 = "2000年第一季度国民总值产业构成分布柱状图"
43     title2 = "2017年第一季度国民总值产业构成分布柱状图"
44     xlabel1 = "产业"
45     xlabel2 = "行业"
46     ticks1 = [tmp[:4] for tmp in columns[3:6]]
47     ticks2 = [tmp[:2] for tmp in columns[6:]]
48     draw_data(1, 3, 5, 4, 0, values, fig, plt, title1, xlabel1, ticks1)
49     draw_data(2, 3, 5, 4, 68, values, fig, plt, title2, xlabel1, ticks1)
50     draw_data(3, 6, 14, 10, 0, values, fig, plt, title1, xlabel2, ticks2)
51     draw_data(4, 6, 14, 10, 68, values, fig, plt, title2, xlabel2, ticks2)
52     save_data()
53 
54 
55 if __name__ == '__main__':
56     main()