Plotly_PyQt5.py

# -*- coding: utf-8 -*-
"""
Created on Thu Feb 16 16:29:55 2017

@author: Administrator
"""
import pandas as pd
import os
import plotly.offline as pyof
import plotly.graph_objs as go

import numpy as np
import matplotlib.pyplot as plt


class Plotly_PyQt5():
	def __init__(self):
		plotly_dir = 'plotly_html'
		if not os.path.isdir(plotly_dir):
			os.mkdir(plotly_dir)
			
		self.path_dir_plotly_html = os.getcwd() + os.sep + plotly_dir
	   
	def get_plotly_path_if_hs300_bais(self,file_name='if_hs300_bais.html'):
		path_plotly = self.path_dir_plotly_html + os.sep + file_name
		df = pd.read_excel(r'if_index_bais.xlsx')
	   
		'''绘制散点图'''
		line_main_price = go.Scatter(
			x=df.index,
			y=df['main_price'],
			name='main_price',
			connectgaps=True, # 这个参数表示允许连接数据缺口
		)
	   
		line_hs300_close = go.Scatter(
			x=df.index,
			y=df['hs300_close'],
			name='hs300_close',
			connectgaps=True,
		)
		data = [line_hs300_close,line_main_price]

		layout = dict(title='if_hs300_bais',
					  xaxis=dict(title='Date'),
					  yaxis=dict(title='Price'),
					  )

		fig = go.Figure(data=data, layout=layout)  
		pyof.plot(fig, filename=path_plotly, auto_open=False)
		return path_plotly
	   
	def get_plot_path_matplotlib_plotly(self,file_name='matplotlib_plotly.html'):
		path_plotly = self.path_dir_plotly_html + os.sep + file_name

		N = 50
		x = np.random.rand(N)
		y = np.random.rand(N)
		colors = np.random.rand(N)
		area = np.pi * (15 * np.random.rand(N)) ** 2  # 0 to 15 point radii
		scatter_mpl_fig = plt.figure()
		plt.scatter(x, y, s=area, c=colors, alpha=0.5)

		pyof.plot_mpl(scatter_mpl_fig, filename=path_plotly, resize=True,auto_open=False)
		return path_plotly