import cufflinks as cf
import seaborn as sns
import plotly as py
import plotly.graph_objs as go
from plotly.offline import download_plotlyjs, init_notebook_mode, plot, iplot
init_notebook_mode(connected=True)
cf.go_offline()

df = sns.load_dataset('iris')
df['species_id'] = df['species'].map({'setosa': 1, 'versicolor': 2, 'virginica': 3})  # 用于颜色映射

py.offline.plot({
    "data": [go.Parcoords(
        line=dict(color=df['species_id'],
                  colorscale=[[0, '#D7C16B'], [0.5, '#23D8C3'], [1, '#F3F10F']]),
        dimensions=list([
            dict(range=[2, 8],
                 constraintrange=[4, 8],
                 label='Sepal Length', values=df['sepal_length']),
            dict(range=[1, 6],
                 label='Sepal Width', values=df['sepal_width']),
            dict(range=[0, 8],
                 label='Petal Length', values=df['petal_length']),
            dict(range=[0, 4],
                 label='Petal Width', values=df['petal_width'])
        ])
    )],
    "layout": go.Layout(title='Iris parallel coordinates plot')
})