高中生比赛求救急急急

朱宝恒

1. import matplotlib.pyplot as plt
   
2. import random
   
3. import pandas as pd
   
4. import pickle
   
5. import matplotlib.ticker as ticker
   
6. import matplotlib.animation as animation
   
7. def randomcolor():#颜色随机生成
   
8.     colorlist = ['1','2','3','4','5','6','7','8','9','A','B','C','D','E','F']
   
9.     color =''
   
10.     for i in range(6):
    
11.         color += random.choice(colorlist)
    
12.     return '#'+ color
    
13. def get_pandafile():
    
14.     data0 = pd.read_excel('china_provincedata.xlsx', sheet_name='china_provincedata',usecols=['provinceShortName','累计确诊','dateId'])
    
15.     pickle_file = open('pdfile.pkl','wb')
    
16.     pickle.dump(data0,pickle_file)
    
17.     pickle_file.close()
    
18. 
    
19. def load_data():
    
20.     pickle_file = open('pdfile.pkl','rb')
    
21.     pdfile = pickle.load(pickle_file)
    
22.     return pdfile
    
23. df=load_data()
    
24. def get_colordict():
    
25. 
    
26. 
    
27.     area_list1 = set(df['provinceShortName'])
    
28.     color_list = []
    
29.     for i in range(len(area_list1)):
    
30.         str_1 = randomcolor()
    
31.         color_list.append(str_1)
    
32.         str_1 = randomcolor()
    
33.     area_list2 = [i for i in area_list1]
    
34. 
    
35.     color_dict = dict(zip(area_list2,color_list))
    
36.     return color_dict
    
37. 
    
38. def illpeople(current_day):
    
39.     fig, ax = plt.subplots(figsize=(15, 8))
    
40.     dff = df[df['dateId'].eq(current_day)].sort_values(by='累计确诊', ascending=True).tail(12)
    
41.     ax.clear()
    
42.     ax.barh(dff['provinceShortName'],dff['累计确诊'],color = [get_colordict()[x] for x in dff['provinceShortName']])
    
43.     dx = dff['累计确诊'].max()/200
    
44.     for i, (value, name) in enumerate(zip(dff['累计确诊'], dff['provinceShortName'])):
    
45.         ax.text(value - dx, i, name, size=14, weight=600, ha='right', va='bottom')
    
46. 
    
47.         ax.text(value + dx, i, f'{value:,.0f}', size=14, ha='left', va='center')
    
48.     ax.text(1, 0.4, current_day, transform=ax.transAxes, color='#777777', size=46, ha='right', weight=800)
    
49.     ax.text(0, 1.06, 'Number of people', transform=ax.transAxes, size=12, color='#777777')
    
50. 
    
51.     # set_major_formatter表示刻度尺格式；
    
52.     ax.xaxis.set_major_formatter(ticker.StrMethodFormatter('{x:,.0f}'))
    
53.     ax.xaxis.set_ticks_position('top')
    
54.     ax.tick_params(axis='x', colors='#777777', labelsize=12)
    
55.     ax.set_yticks([])
    
56.     # margins表示自动缩放余额；
    
57.     ax.margins(0, 0.01)
    
58.     # 设置后面的网格
    
59.     ax.grid(which='major', axis='x', linestyle='-')
    
60.     # 刻度线和网格线是在图标上方还是下方，True为下方
    
61.     ax.set_axisbelow(True)
    
62.     ax.text(0, 1.15, 'Number of people infected with the COVID-19',
    
63.             transform=ax.transAxes, size=24, weight=600, ha='left', va='top')
    
64.     ax.text(1, 0, 'by@Baoheng Zhu', transform=ax.transAxes, color='#777777', ha='right',
    
65.             bbox=dict(facecolor='white', alpha=0.8, edgecolor='white'))
    
66.     # 取消图表周围的方框显示
    
67.     plt.box(False)
    
68. 
    
69. fig, ax = plt.subplots(figsize=(15, 8))
    
70. 
    
71. 
    
72. 
    
73. 
    
74. if __name__ == '__main__':
    
75.     randomcolor()
    
76.     get_pandafile()
    
77.     load_data()
    
78.     get_colordict()
    
79.     fig=illpeople(20200122)
    
80.     animator = animation.FuncAnimation(fig, illpeople, frames=range(20200122, 20200608))
    
81.     plt.show()
    

复制代码

如题，省里组织python竞赛，说可以借鉴他人代码，我就在网上找了一个课题，自己改编，但是赛事要求只能用matplotlib
库，我要做的是新冠疫情动态图，没有ipython，没办法动，只能呈现一张图片，各位大神快来帮帮我吧，怎么才能用matploblib自身显示动画呢，跪谢

Twilight6

你参考这个代码吧，我也是去隔壁度娘搬运来的，这个就算动态的图了：

1. import matplotlib.pyplot as plt
   
2. import numpy as np
   
3. 
   
4. ax=[]   #保存图1数据
   
5. ay=[]
   
6. bx=[]   #保存图2数据
   
7. by=[]
   
8. num=0   #计数
   
9. plt.ion()    # 开启一个画图的窗口进入交互模式，用于实时更新数据
   
10. # plt.rcParams['savefig.dpi'] = 200 #图片像素
    
11. # plt.rcParams['figure.dpi'] = 200 #分辨率
    
12. plt.rcParams['figure.figsize'] = (10, 10)        # 图像显示大小
    
13. plt.rcParams['font.sans-serif']=['SimHei']   #防止中文标签乱码，还有通过导入字体文件的方法
    
14. plt.rcParams['axes.unicode_minus'] = False
    
15. plt.rcParams['lines.linewidth'] = 0.5   #设置曲线线条宽度
    
16. while num<100:
    
17.     plt.clf()    #清除刷新前的图表，防止数据量过大消耗内存
    
18.     plt.suptitle("总标题",fontsize=30)             #添加总标题，并设置文字大小
    
19.     g1=np.random.random()  #生成随机数画图
    
20. #图表1
    
21.     ax.append(num)      #追加x坐标值
    
22.     ay.append(g1)       #追加y坐标值
    
23.     agraphic=plt.subplot(2,1,1)
    
24.     agraphic.set_title('子图表标题1')      #添加子标题
    
25.     agraphic.set_xlabel('x轴',fontsize=10)   #添加轴标签
    
26.     agraphic.set_ylabel('y轴', fontsize=20)
    
27.     plt.plot(ax,ay,'g-')                #等于agraghic.plot(ax,ay,'g-')
    
28. #图表2
    
29.     bx.append(num)
    
30.     by.append(g1)
    
31.     bgraghic=plt.subplot(2, 1, 2)
    
32.     bgraghic.set_title('子图表标题2')
    
33.     bgraghic.plot(bx,by,'r^')
    
34. 
    
35.     plt.pause(0.4)     #设置暂停时间，太快图表无法正常显示
    
36.     if num == 15:
    
37.         plt.savefig('picture.png', dpi=300)  # 设置保存图片的分辨率
    
38.         #break
    
39.     num=num+1
    
40. 
    
41. plt.ioff()       # 关闭画图的窗口，即关闭交互模式
    
42. plt.show()       # 显示图片，防止闪退
    

复制代码

java2python

你的程序想表达的数据，可能比较复杂，真的不懂，下面代码是反应每天前五位省份的数据变动情况的（当然由于其他省份和湖北比比例很小，看上去变动不明显）：

1. import matplotlib.pyplot as plt
   
2. import numpy as np
   
3. from matplotlib import animation
   
4. import pickle
   
5. 
   
6. def load_data():
   
7.     pickle_file = open('pdfile.pkl','rb')
   
8.     pdfile = pickle.load(pickle_file)
   
9.     return pdfile
   
10. 
    
11. start_day = 20200122
    
12. month_day=[0,31,29,31,30,31,30]
    
13. def all_day(start,end):
    
14.     days = []
    
15.     cur_day = start
    
16.     while cur_day < end:
    
17.         mon = (cur_day//100) % 100
    
18.         day = cur_day % 100
    
19.         if day == month_day[mon]:
    
20.             mon += 1
    
21.             day = 1
    
22.         else:
    
23.             day +=1
    
24.         cur_day = 20200000+mon*100+day
    
25.         days.append(cur_day)
    
26.     return days
    
27. days = all_day(20200122,20200608)
    
28. 
    
29. fig=plt.figure(1,figsize=(4,3))
    
30. ax=fig.add_subplot(111)
    
31. ax.set_title('bar_animate_test')
    
32. ax.set_xlabel('xlable')
    
33. N=5
    
34. frames=len(days)
    
35. x=np.arange(1,N+1)
    
36. 
    
37. df=load_data()
    
38. collection=[]
    
39. collection.append([i for i in x])
    
40. for i in range(frames):
    
41.     dff = df[df['dateId'].eq(days[i])].sort_values(by='confirmedCount', ascending=True).tail(5)
    
42.     ax.clear()
    
43.     c=[]
    
44.     for l, (value, name) in enumerate(zip(dff['confirmedCount'], dff['provinceShortName'])):
    
45.         c.append(value)
    
46. 
    
47.     collection.append(c)
    
48. print(collection)
    
49. xstd=[0,1,2,3,4]
    
50. bars=ax.bar(x,collection[0],0.30)
    
51. def animate(fi):
    
52.     ax.set_ylim(0,max(collection[fi])+3)
    
53.     for rect ,yi in zip(bars,collection[fi]):
    
54.         rect.set_height(yi)
    
55.     return bars
    
56. anim=animation.FuncAnimation(fig,animate,frames=frames,interval=300,repeat=False)
    
57. plt.show()

复制代码

java2python

你的程序，应该是拷贝Bar Chart Race（条形竞赛图） in Python with Matplotlib
https://zhuanlan.zhihu.com/p/94331647
可能他也发表在别的网站，但作者就是他。以下代码是能够动态表示的，只是有乱码

1. #导入工具包
   
2. import pandas as pd
   
3. import matplotlib.pyplot as plt
   
4. import matplotlib.ticker as ticker
   
5. import matplotlib.animation as animation
   
6. #from IPython.display import HTML
   
7. import random
   
8. 
   
9. df = pd.read_csv('china_provincedata.csv', usecols=['provinceName', 'provinceShortName', 'dateId', 'confirmedCount'])
   
10. 
    
11. fig, ax = plt.subplots(figsize=(15, 8))
    
12. 
    
13. colors = dict(zip(
    
14.     ['India','Europe','Asia','Latin America','Middle East','North America','Africa'],
    
15.     ['#adb0ff', '#ffb3ff', '#90d595', '#e48381', '#aafbff', '#f7bb5f', '#eafb50']
    
16. ))
    
17. group_lk = df.set_index('provinceName')['provinceShortName'].to_dict()
    
18. def draw_barchart(current_day):
    
19.     dff = df[df['dateId'].eq(current_day)].sort_values(by='confirmedCount', ascending=True).tail(10)
    
20.     ax.clear()
    
21.     #ax.barh(dff['provinceName'], dff['confirmedCount'], color=[colors[group_lk[x]] for x in dff['provinceName']])
    
22.     ax.barh(dff['provinceShortName'],dff['confirmedCount'],color = [get_colordict()[x] for x in dff['provinceShortName']])
    
23.     dx = dff['confirmedCount'].max() / 200
    
24.     for i, (value, name) in enumerate(zip(dff['confirmedCount'], dff['provinceName'])):
    
25.         ax.text(value-dx, i,     name,           size=14, weight=600, ha='right', va='bottom')
    
26.         ax.text(value-dx, i-.25, group_lk[name], size=10, color='#444444', ha='right', va='baseline')
    
27.         ax.text(value+dx, i,     f'{value:,.0f}',  size=14, ha='left',  va='center')
    
28.     # ... 优化风格
    
29.     ax.text(1, 0.4, current_day, transform=ax.transAxes, color='#777777', size=46, ha='right', weight=800)
    
30.     ax.text(0, 1.06, 'Population (thousands)', transform=ax.transAxes, size=12, color='#777777')
    
31.     ax.xaxis.set_major_formatter(ticker.StrMethodFormatter('{x:,.0f}'))
    
32.     ax.xaxis.set_ticks_position('top')
    
33.     ax.tick_params(axis='x', colors='#777777', labelsize=12)
    
34.     ax.set_yticks([])
    
35.     ax.margins(0, 0.01)
    
36.     ax.grid(which='major', axis='x', linestyle='-')
    
37.     ax.set_axisbelow(True)
    
38.     ax.text(0, 1.12, 'The most populous cities in the world from 1500 to 2018',
    
39.             transform=ax.transAxes, size=24, weight=600, ha='left')
    
40.     ax.text(1, 0, 'by @pratapvardhan; credit @jburnmurdoch', transform=ax.transAxes, ha='right',
    
41.             color='#777777', bbox=dict(facecolor='white', alpha=0.8, edgecolor='white'))
    
42.     plt.box(False)
    
43.    
    
44. month_day=[0,31,29,31,30,31,30]
    
45. def all_day(start,end):
    
46.     days = []
    
47.     cur_day = start
    
48.     while cur_day < end:
    
49.         mon = (cur_day//100) % 100
    
50.         if cur_day % 100 == month_day[mon]:
    
51.             cur_day = 20200000+(mon+1)*100+1
    
52.         else:
    
53.             cur_day += 1
    
54.         days.append(cur_day)
    
55.     return days
    
56. days = all_day(20200122,20200608)
    
57. 
    
58. def randomcolor():#颜色随机生成
    
59.     colorlist = ['1','2','3','4','5','6','7','8','9','A','B','C','D','E','F']
    
60.     color =''
    
61.     for i in range(6):
    
62.         color += random.choice(colorlist)
    
63.     return '#'+ color
    
64. 
    
65. def get_colordict():
    
66.     area_list1 = set(df['provinceShortName'])
    
67.     color_list = []
    
68.     for i in range(len(area_list1)):
    
69.         str_1 = randomcolor()
    
70.         color_list.append(str_1)
    
71.         str_1 = randomcolor()
    
72.     area_list2 = [i for i in area_list1]
    
73. 
    
74.     color_dict = dict(zip(area_list2,color_list))
    
75.     return color_dict
    
76. 
    
77. draw_barchart(20200607)
    
78. 
    
79. animator = animation.FuncAnimation(fig, draw_barchart, frames=days)
    
80. #HTML(animator.to_jshtml())
    
81. plt.show()
    

复制代码

朱宝恒

java2python 发表于 2020-6-20 13:01
你的程序，应该是拷贝Bar Chart Race（条形竞赛图） in Python with Matplotlib
https://zhuanlan.zhihu.c ...

对，就是这个，但是比赛没有ipython库没法动态显示，我想解决的是这个问题

java2python

朱宝恒 发表于 2020-6-20 18:12
对，就是这个，但是比赛没有ipython库没法动态显示，我想解决的是这个问题

我上边的程序没用到这个库啊，可以动态表示的啊，实际只表示条形图，不用matlab，用pygame也可以做到，只不过就是矩形嘛。你原先程序range(20200122,20200608)，这个错了，走不动了，不会进入循环了。这个日期的列表需要手动生成。其他应该没问题的。

java2python

不太好意思贴，gif文件太大，只能表示几天的，你这个区间有几个月：

896944660

巧了，我也才参加省里的比赛，也用的是matplotlib，不过我还打算深入切PyQt开发，

朱宝恒

Traceback (most recent call last):
  File "E:\python\省赛材料\text3.py", line 96, in <module>
    draw_barchart(20200607)
  File "E:\python\省赛材料\text3.py", line 37, in draw_barchart
    ax.text(value - dx, i - .25, group_lk[name], size=10, color='#444444', ha='right', va='baseline')
KeyError: '山东省'
我运行了一下你的代码，为什么山东省会出问题啊

java2python

这个我也遇到了，怎么解决的，真的想不起来了，因为程序改的地方太多了：其实不算多，要改的地方A,B,C,D,E
结果A改成A+,B改成B+，C改成C+，如此组合起来，有时自己也晕了。真的想不起来了。

java2python

别自己再一定要搞PKL了，泡菜，csv太低端？虽然这里毫无问题，其实没明白为何一定要excel线路，泡菜，怎么麻烦。或许是拿了别人的程序，觉得不加一点不好意思。怎么说呢，一开始应该先做能跑通的，看你好像，调试不咋过关，遇到问题，就求助了。不是说求助不行。而是毕竟这一来一回，哦特慢。。。
先搞一个能跑的，这个是最关键的，然后一步一步加，因为你加了三步，结果又不动了，就搞不清楚是哪一步出问题了。
调试很关键，为啥，毕竟不可能什么都求助，把程序弄成能跑是最基础的。
比如keyerror，可能是group_lk[name],找不到“山东省”的对应数据，实际我因为忘了。你最好能够有调试环境，光标行，用鼠标指上去，看看里面什么值，什么情况，一下子就明白了。
贴了个附件，这个应该是能跑的，如果你不加入自己的那些复杂东西的话。要加也是一步一步加，不是整体工程全搞完，然后一看不行，这有点扯谈。

朱宝恒

谢谢你了，已经调试好了，没传上来，麻烦你了，用泡菜的原因其实是不让用pandas库，我原来用正则都写好了，又看到你的代码，然后我寻思着用二进制保存到本地然后一起上交，没有别的意思
谢谢你的帮助，其实我特别不愿意麻烦你们，我以后会继续努力的

朱宝恒

哦错了，不是你这个，就是我原来拷贝的那个