Easy Plotly

easy_pltly.py

import plotly as py
import plotly.graph_objs as go
import numpy as np

#py.tools.set_credentials_file(username='user_name', api_key='api_key')
py.offline.init_notebook_mode(connected=True)

def make_trace(x, y, mode='markers', pltname=None, shape='spline'):
    trace = go.Scatter(x = x, y = y, mode = mode, name = pltname, line = dict(shape=shape))
    return trace

def plotly_plot(traces, title=None, xtitle=None, ytitle=None, mode='markers', pltname=None):
    layout = go.Layout(title = title, yaxis = dict(title = ytitle), xaxis = dict(title = xtitle))
    fig = go.Figure(data=traces, layout = layout)
    py.offline.iplot(fig)

if __name__ == '__main__':
    x = np.linspace(0, np.pi*2, 200)
    y = np.sin(x)

    trace1 = make_trace(x, y, mode='lines')
    trace2 = make_trace(x, y*2, mode='lines')
    trace3 = make_trace(x, y/2, mode='lines')

    traces = [trace1, trace2, trace3]
    plotly_plot(traces)