python学习之小游戏：解析API，生成疫情词云

课程视频链接

跟着麦叔学Python：https://b23.tv/Yx4Eyq

1. 代码

import requests
import json

# 1.请求数据
url = 'https://api.yimian.xyz/coro'
response = requests.get(url)
print(response)

data = requests.get(url).text
print(data)

# 2.解析数据-转换成JSON对象
city_dict = {}  # 做成字典
json_data = json.loads(data)
print(len(json_data))

for p in json_data:
    if 'cities' in p:
        for c in p['cities']:
            cityName = c['cityName']
            cCount = c['confirmedCount']
            sCount = c['suspectedCount']
            total = int(cCount) + int(sCount)
            city_dict[cityName] = total
    else:
        cityName = p['provinceName']
        cCount = p['confirmedCount']
        sCount = p['suspectedCount']
        total = int(cCount) + int(sCount)
        city_dict[cityName] = total
print(city_dict)

# 3.写入文件
with open('city_count.txt', 'w') as f:
    for k, v in city_dict.items():
        f.write('{}:{}\n'.format(k,str(v)))

# 4.生成词云
import wordcloud
import matplotlib.pyplot as plt

# ccloud = wordcloud.WordCloud(background_color='white',width=1600,height=800,font_path = 'C:\Windows\Fonts\STXINGKA.TTF')
# ccloud.generate_from_frequencies(frequencies = city_dict)
#
# plt.figure()
# plt.imshow(ccloud,interpolation='bilinear')
# plt.axis('off')
# plt.show()

# 5.生成爱心词云，并存图片
import numpy as np
from PIL import Image
heart_mask =np.array(Image.open('heart.jpg'))
ccloud = wordcloud.WordCloud(background_color='white',mask=heart_mask,font_path = 'C:\Windows\Fonts\STXINGKA.TTF')
ccloud.generate_from_frequencies(frequencies = city_dict)

plt.figure(dpi=1200) #指定图片清晰度
plt.imshow(ccloud,interpolation='bilinear')
plt.axis('off')
# plt.show()
plt.savefig('heart.png') #保存

2. 生成的词云图片