222 lines
4.9 KiB
Python
222 lines
4.9 KiB
Python
#!/usr/bin/env python
|
|
# coding: utf-8
|
|
|
|
# In[1]:
|
|
|
|
|
|
import pandas as pd
|
|
import datetime
|
|
import numpy as np
|
|
import plotly as pl
|
|
import plotly.graph_objs as go
|
|
import matplotlib.pyplot as plt
|
|
import random
|
|
|
|
|
|
import datetime
|
|
import matplotlib.dates as mdates
|
|
import matplotlib.pyplot as plt
|
|
import plotly
|
|
import plotly.graph_objs as go
|
|
from plotly.offline import init_notebook_mode, iplot
|
|
from plotly.subplots import make_subplots
|
|
init_notebook_mode()
|
|
|
|
#import CoreTraidMath
|
|
import plotly.express as px
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# In[ ]:
|
|
|
|
|
|
|
|
|
|
|
|
# In[2]:
|
|
|
|
|
|
class agrigateFig():
|
|
|
|
def __init__(self,data=[],needDraw=False ,subplot_titles=None):
|
|
self.data=data
|
|
self.ans=self.getAgrPlt()
|
|
if needDraw:
|
|
self.subplot_titles=subplot_titles
|
|
self.fig=coreDraw(self.ans,True,self.subplot_titles)
|
|
|
|
|
|
def getAgrPlt(self):
|
|
count=0
|
|
ans=[]
|
|
for i in self.data:
|
|
count=count+1
|
|
if type(i)==list:
|
|
for g in i:
|
|
for j in g.figDict:
|
|
ans.append(j)
|
|
ans[-1]['row']=count
|
|
else:
|
|
for j in i.figDict:
|
|
ans.append(j)
|
|
ans[-1]['row']=count
|
|
return ans
|
|
|
|
|
|
# In[3]:
|
|
|
|
|
|
class corePlt():
|
|
def __init__(self, params={
|
|
'vtype':'',
|
|
'df':pd.DataFrame(),
|
|
'row':1,
|
|
'col':1,
|
|
'name':''
|
|
}):
|
|
self.vtype=params['vtype']
|
|
self.df=params['df']
|
|
self.row=params['row']
|
|
self.col=params['col']
|
|
self.name=params['name']
|
|
if 'colorType' in params.keys():
|
|
self.colorType=params['colorType']
|
|
|
|
|
|
|
|
|
|
|
|
class coreDraw():
|
|
def __init__(self, data=[],needShow=False,subplot_titles={}):
|
|
self.data=self.getPlts(data)
|
|
self.needShow=needShow
|
|
self.subplot_titles=subplot_titles
|
|
self.ans=self.getAns()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def getBarColorList(self,l,colorType):
|
|
if colorType=='diffAbs':
|
|
ans=['green']
|
|
for i in range(1,len(l)):
|
|
if abs(l[i])>abs(l[i-1]):
|
|
ans.append('green')
|
|
else:
|
|
ans.append('red')
|
|
elif colorType=='diff':
|
|
ans=['green']
|
|
for i in range(1,len(l)):
|
|
if (l[i])>(l[i-1]):
|
|
ans.append('green')
|
|
else:
|
|
ans.append('red')
|
|
elif colorType=='normal':
|
|
ans=[]
|
|
for i in range(len(l)):
|
|
ans.append('gray')
|
|
return ans
|
|
|
|
def getPlts(self, data):
|
|
ans=None
|
|
|
|
if type(data)==list:
|
|
ans=[]
|
|
for i in data:
|
|
ans.append(corePlt(i))
|
|
else:
|
|
ans=[corePlt(data)]
|
|
|
|
|
|
|
|
|
|
return ans
|
|
|
|
def getAns(self):
|
|
'''
|
|
data list
|
|
vtype
|
|
df
|
|
row=1
|
|
col=1
|
|
name
|
|
|
|
|
|
|
|
'''
|
|
|
|
ans=None
|
|
|
|
|
|
|
|
|
|
maxRow=1
|
|
maxCol=1
|
|
for i in self.data:
|
|
if i.row > maxRow:
|
|
maxRow =i.row
|
|
if i.col > maxCol:
|
|
maxCol =i.col
|
|
|
|
fig = make_subplots(
|
|
rows=maxRow,
|
|
cols=maxCol,
|
|
shared_xaxes=True,
|
|
vertical_spacing=0.1,
|
|
shared_yaxes=True,
|
|
#horizontal_spacing=0.02,
|
|
#column_widths=[]
|
|
subplot_titles=self.subplot_titles
|
|
)
|
|
|
|
|
|
fig.update_layout(xaxis_rangeslider_visible=False)
|
|
fig.update_layout(barmode='relative')
|
|
|
|
for i in self.data:
|
|
if i.vtype=='Scatter':
|
|
fig.add_trace(go.Scatter(x=i.df['date'],y=i.df['value'],name=i.name), row=i.row, col=i.col)
|
|
elif i.vtype=='OCHL':
|
|
fig.add_trace(go.Candlestick(
|
|
x=i.df['date'],
|
|
open=i.df['open'],
|
|
high=i.df['high'],
|
|
low=i.df['low'],
|
|
close=i.df['close'],
|
|
name=i.name),
|
|
row=i.row, col=i.col
|
|
)
|
|
elif i.vtype=='Bars':
|
|
for j in i.df.keys():
|
|
if j!='date':
|
|
try:
|
|
colorType=i.colorType
|
|
except:
|
|
colorType='normal'
|
|
colors=self.getBarColorList(i.df[j],colorType)
|
|
fig.add_trace(go.Bar(x=i.df['date'], y=i.df[j],name=j,marker_color=colors),row=i.row, col=i.col)
|
|
|
|
|
|
|
|
|
|
ans=fig
|
|
if self.needShow:
|
|
plotly.offline.iplot(fig)
|
|
return ans
|
|
|
|
|
|
# In[ ]:
|
|
|
|
|
|
|
|
|