import plotly
import plotly.plotly as py
import numpy as np
plotly.offline.init_notebook_mode()
exp = 2.71
N = 3
x_start = -N
x_end = N
dx = 0.1
y_start = -N
y_end = N
dy = 0.1
x_axis = np.arange(x_start, x_end, dx)
y_axis = np.arange(y_start, y_end, dy)
def pit(offset, x):
pit1 = []
for position in x:
pit1.append(1 - exp**(-36 * (position - offset)**2))
return pit1
def v_x(x):
vx = exp**(-x**2)
return vx
def v_y(x):
vy = 0.5 * exp**(-x**2)
return vy
def density(vx, vy):
den = []
for v1 in vx:
row = []
for v2 in vy:
row.append(v1 * v2)
den.append(row)
return den
vx = v_x(x_axis)
vy = v_y(y_axis)
den = density(vx, vy)
def contour(step=None):
return dict(
type='contour',
z=den,
colorscale=[[0, 'rgb(255,255,255)'], [1, 'rgb(49,163,84)']],
x=x_axis,
y=y_axis,
)
def vy_projection(step):
return dict(
visible=True,
type='scatter',
name=str(step),
marker=dict(color='rgb(255,0,0)'),
yaxis='y2',
x=x_axis,
y=vx * pit(step, x_axis)
)
def vx_projection(step):
return dict(
visible=True,
type='scatter',
name=str(step),
marker=dict(color='rgb(255,0,255)'),
xaxis='x2',
x=vy,
y=y_axis
)
trace1 = [
[vy_projection(step), vx_projection(step)]
for step in np.arange(-3, 3, 0.5)]
for plot in trace1[-1]:
plot['visible'] = True
steps = []
for i in range(len(trace1)):
step = dict(
method='restyle',
args=['visible', [False] * len(trace1)],
)
step['args'][1][i] = True
steps.append(step)
sliders = [dict(
active=10,
currentvalue={"prefix": "Step: "},
pad={"t": len(trace1)},
steps=steps
)]
layout = dict(
autosize=False,
width=500,
height=500,
sliders=sliders,
xaxis=dict(
range=[-3, 3],
),
xaxis2=dict(
domain=[0.9, 1],
showgrid=False,
zeroline=False,
),
yaxis=dict(
range=[-3, 3],
),
yaxis2=dict(
domain=[0.9, 1],
showgrid=False,
zeroline=False,
)
)
data = trace1[0]
fig = go.Figure(data=data, layout=layout)
# plotly.offline.plot(fig, filename='manipulate.html')
iplot(fig, filename='Sine Wave Slider')