当前位置 : 主页 > 编程语言 > python >

Python实战系列 | 正则提取数据并绘图

来源:互联网 收集:自由互联 发布时间:2023-08-10
本期给大家分享 《Python实战系列》 的第一篇文章:正则提取数据并绘图,该系列主要来自粉丝的实际问题,后期会不断更新,希望对你有所帮助,如有疑问或者需要改进的地方可以私


本期给大家分享《Python实战系列》的第一篇文章:正则提取数据并绘图,该系列主要来自粉丝的实际问题,后期会不断更新,希望对你有所帮助,如有疑问或者需要改进的地方可以私信小编。
任务描述: 提取txt文件各语言数据绘制折线图,data.txt文件截图如下(太长了直截了一部分):

Python实战系列 | 正则提取数据并绘图

乍一看很像json格式的文件,实际上不是Python实战系列 | 正则提取数据并绘图
进入正题:

?️‍? 1. 读取数据

with open('data.txt') as f:
    data = f.read()
?️‍? 2. 正则提取花括号{}里面的信息
datas = re.findall('({.*?})',data)
Python实战系列 | 正则提取数据并绘图

Python实战系列 | 正则提取数据并绘图

内容基本上就是编程语言各日期的占比情况,接下来就是提取日期信息和数据信息了。


?️‍? 3. 提取name,注意转义(')
Python实战系列 | 正则提取数据并绘图
re.findall('\'(.*)\'',datas[0])[0]
?️‍? 4. 提取日期和数据,注意数据小数点
re.findall('(\d+(\.\d+)?)',datas[0])
每4个数据为一组,所以做个step=4的循环即可:
for i in range(0,len(datas_tmp),4):
    datas_f.append(float(datas_tmp[i+3][0]))
    dates_f.append(f'{datas_tmp[i][0]}-{datas_tmp[i+1][0]}-{datas_tmp[i+2][0]}')

?️‍? 5. 数据提取部分完整代码

# 处理数据
with open('data.txt') as f:
    data = f.read()
datas = re.findall('({.*?})',data)
names = []
dates_result = []
datas_result = []
for idx,dd in enumerate(datas):
    datas_f = []
    dates_f = []
    name = re.findall('\'(.*)\'',dd)[0]
    names.append(name)
    datas_tmp = re.findall('(\d+(\.\d+)?)',dd)
    for i in range(0,len(datas_tmp),4):
        datas_f.append(float(datas_tmp[i+3][0]))
        dates_f.append(f'{datas_tmp[i][0]}-{datas_tmp[i+1][0]}-{datas_tmp[i+2][0]}')
    datas_result.append(datas_f)
    dates_result.append(dates_f)


?️‍? 6. 绘图

绘图部分直接用matplotlib的plot循环绘制即可,代码如下:

# 绘图
plt.figure(figsize=(20, 10), dpi=100)
for i in range(len(names)):
    plt.plot(dates_result[i], datas_result[i], label=names[i])
ax = plt.gca()
ax.xaxis.set_major_locator(ticker.MultipleLocator(20))
plt.ylabel("Ratings(%)", fontdict={'size': 16})
plt.title("TIOBE Programming Community Index", fontdict={'size': 20})
plt.legend(loc='best')
plt.grid(True, linestyle='--', alpha=0.5)
plt.show()

Python实战系列 | 正则提取数据并绘图


网友评论