coding=gbk
import json
import pygal
import math
from itertools import groupby
def draw_line(x_data, y_data, title, y_legend):
xy_map = []
“zip() 函数用于将可迭代的对象作为参数,将对象中对应的元素打包成一个个元组,然后返回由这些元组组成的列表”
“sorted() 函数对所有可迭代的对象进行排序操作,默认升序”
“groupby()函数主要作用是进行数据的分组以及分组后地组内运算”
for x, y in groupby(sorted(zip(x_data, y_data)), key=lambda :[0]):
y_list = [v for _, v in y]
xy_map.append([x, sum(y_list)/len(y_list)])
“*zip()函数是zip()函数的逆过程,将zip对象变成原先组合前的数据”
x_unique, y_mean = [*zip(*xy_map)]
line_chart = pygal.Line()
line_chart.title = title
line_chart.x_labels = x_unique
line_chart.add(y_legend, y_mean)
line_chart.render_to_file(title+’.svg’)
return line_chart
将数据加载到一个列表中
filename = ‘btc_close_2017.json’
with open(filename) as f:
btc_data = json.load(f)
dates = []
months = []
weeks = []
weekdays = []
close = []
打印每一天的信息
for btc_dict in btc_data:
dates.append(btc_dict[‘date’])
months.append(int(btc_dict[‘month’]))
weeks.append(int(btc_dict[‘week’]))
weekdays.append(btc_dict[‘weekday’])
close.append(int(float(btc_dict[‘close’])))
绘图收盘价折线图
line_chart = pygal.Line(x_label_rotation=20, show_minor_x_labels=False)
line_chart.title = ‘收盘价(¥)’
line_chart.x_labels = dates
N = 20 # x轴坐标每隔20天显示一次
line_chart.x_labels_major = dates[::N]
line_chart.add(‘收盘价’, close)
line_chart.render_to_file(‘收盘价折线图(¥).svg’)
绘图收盘价折线图
line_chart = pygal.Line(x_label_rotation=20, show_minor_x_labels=False)
line_chart.title = ‘收盘价对数变换(¥)’
line_chart.x_labels = dates
N = 20 # x轴坐标每隔20天显示一次
line_chart.x_labels_major = dates[::N]
close_log = [math.log10(i) for i in close]
line_chart.add(‘log收盘价’, close_log)
line_chart.render_to_file(‘收盘价对数变换折线图(¥).svg’)
绘图收盘价月日均值
idx_month = dates.index(‘2017-12-01’)
line_chart_month = draw_line(months[:idx_month], close[:idx_month], ‘收盘价月日均值(¥)’, ‘月日均值’)
line_chart_month
绘图收盘价周日均值
idx_week = dates.index(‘2017-12-11’)
line_chart_week = draw_line(weeks[1:idx_week], close[1:idx_week], ‘收盘价周日均值(¥)’, ‘周日均值’)
line_chart_week
绘制收盘价星期均值
idx_weekday = dates.index(‘2017-12-11’)
wd = [‘Monday’, ‘Tuesday’, ‘Wednesday’, ‘Thursday’, ‘Friday’, ‘Saturday’, ‘Sunday’]
weekdays_int = [wd.index(w) + 1 for w in weekdays[1:idx_weekday]]
line_chart_weekday = draw_line(weekdays_int, close[1:idx_weekday], ‘收盘价星期均值(¥)’, ‘星期均值’)
line_chart_weekday.x_labels = [‘周一’, ‘周二’, ‘周三’, ‘周四’, ‘周五’, ‘周六’, ‘周日’]
line_chart_weekday.render_to_file(‘收盘价星期均值(¥).svg’)
收盘价数据仪表盘
with open(‘收盘价Dashboard.html’, ‘w’, encoding=‘utf8’) as html_file:
html_file.write(‘
for svg in [‘收盘价折线图(¥).svg’, ‘收盘价对数变换折线图(¥).svg’,‘收盘价月日均值(¥).svg’, ‘收盘价周日均值(¥).svg’, ‘收盘价星期均值(¥).svg’]:
html_file.write(’\n’.format(svg))
html_file.write(’’)