实例4·股票数据分析(pygal,json格式数据处理)


import json
import requests
import pygal
import math
from itertools import groupby

# 将数据加载到一个列表中
filename = 'C://Users/Administrator/Desktop/btc_close_2017.json'
with open(filename) as f:
    texts = json.load(f)  # 1

dates = []
months = []
weeks = []
weekdays = []
close = []
# 每一天的信息
for text in texts:
    dates.append(text['date'])
    months.append(int(text['month']))
    weeks.append(int(text['week']))
    weekdays.append(text['weekday'])
    close.append(int(float(text['close'])))

# 1·收盘价折线图
line_chart = pygal.Line(x_label_rotation=20, show_minor_x_labels=False)  # x轴的标签顺时针旋转20度;不用显示全部标签
line_chart.title = '收盘价(¥)'
line_chart.x_labels = dates  # 设置x轴标签
N = 20  # x轴坐标每隔20天显示一次
line_chart.x_labels_major = dates[::N]  # ②
line_chart.add('收盘价', close)  # 交互添加
line_chart.render_to_file('收盘价折线图(¥).svg')

# 2·收盘价对数变换折线图
line_chart = pygal.Line(x_label_rotation=20, show_minor_x_labels=False)
line_chart.title = '收盘价对数变换(¥)'
line_chart.x_labels = dates
N = 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')


# 3·月日均
def draw_line(x_data, y_data, title, y_legend):  # 传入月份列表,每日收盘数据列表
    xy_map = []
    for x, y in groupby(sorted(zip(x_data, y_data)), key=lambda m: m[0]):  # 合并传入的两数据,并按月排序;之后按照月份分为11个组
        # print(x,y)
        # 1 < itertools._grouper object at 0x00000000048ACA90 >
        # 2 < itertools._grouper object at 0x00000000048ACB00 >
        # ......
        # 11 < itertools._grouper object at 0x00000000048ACB00 >
        y_list = [v for _, v in y]
        xy_map.append([x, sum(y_list) / len(y_list)])  # 计算日平均
    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')
idx_month = dates.index('2017-12-01')  # 返回索引标号,即第334个字典,data是
line_chart_month = draw_line(months[:idx_month], close[:idx_month], '收盘价月日均值(¥)', '月日均值')

#4· 周日均
idx_week = dates.index('2017-12-11')
line_chart_week = draw_line(
    weeks[1:idx_week], close[1:idx_week], '收盘价周日均值(¥)', '周日均值')


# 5·天日均
def draw_line1(x_data, y_data, title, y_legend):
    xy_map = []
    for x, y in groupby(sorted(zip(x_data, y_data)), key=lambda m: m[0]):
        # print(x,y)
        # 1 < itertools._grouper object at 0x00000000048ACA90 >
        # 2 < itertools._grouper object at 0x00000000048ACB00 >
        # ......
        # 11 < itertools._grouper object at 0x00000000048ACB00 >
        y_list = [v for _, v in y]
        xy_map.append([x, sum(y_list) / len(y_list)])  # 计算日平均
    x_unique, y_mean = [*zip(*xy_map)]
    line_chart = pygal.Line()
    line_chart.title = title
    line_chart.x_labels = ['周一', '周二', '周三', '周四', '周五', '周六', '周日']
    line_chart.add(y_legend, y_mean)
    line_chart.render_to_file('收盘价星期均值(¥).svg')
idx_week = 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_week]]
line_chart_weekday = draw_line1(weekdays_int, close[1:idx_week], '收盘价星期均值(¥)', '星期均值')

# html生成
with open('收盘价Dashboard.html', 'w', encoding='utf8') as html_file:
    html_file.write(
        '<html><head><title>收盘价Dashboard</title><meta charset="utf-8"></head><body>\n')
    for svg in [
        '收盘价折线图(¥).svg', '收盘价对数变换折线图(¥).svg', '收盘价月日均值(¥).svg',
        '收盘价周日均值(¥).svg', '收盘价星期均值(¥).svg'
    ]:
        html_file.write(
            '    <object type="image/svg+xml" data="{0}" height=500></object>\n'.format(svg))
    html_file.write('</body></html>')

结果展示
在这里插入图片描述

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值