读取CSV格式文件,修改 echarts 模板数据

需要做一个散点图,但是excel实现的效果不太满意

于是找到了 echarts 找个JS插件,稍微修改了模板文件,得到需要的散点图

以下是读取文件数据,修改模板中data的代码

# -*- coding: utf-8 -*-
"""
Created on Wed Oct 17 14:52:50 2018

@author: sofeien
"""
import os
import re

path= os.path.abspath('散点图.csv')

data_str='var data=['               
with open(path,'r',encoding='utf-8') as f_input:
        line_index=0
        for line in f_input:
            if(line_index==0):
                line_index+=1
                continue
            list_line=[x.strip() for x in re.split('[,\s]',line)]
            data_str += "[%s,%s,'%s']," % (list_line[1],list_line[2],list_line[0])
        data_str = data_str[:-1]+'];'
os.rename('scatter-weight.html', 'scatter-weight_bak.html')
with open('scatter-weight_bak.html','r',encoding='utf-8') as f:
    with open('scatter-weight.html','w',encoding='utf-8') as f_output:
        for line in f:
            if(line.find('var data=[')!=-1):
                f_output.write(data_str)
                f_output.write('\n')
            else:
                f_output.write(line)
os.remove('scatter-weight_bak.html')

猜你喜欢

转载自blog.csdn.net/sofeien/article/details/83112115