1. 程式人生 > >讀取CSV格式檔案,修改 echarts 模板資料

讀取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')