Miguel Angel Ortiz Marin
01/25/2023, 7:46 PMMassinissa Saïdi
01/25/2023, 8:14 PMfrom kedro.extras.datasets.plotly import PlotlyDataSet
import plotly.express as px
import pandas as pd
df_data = pd.DataFrame([[0, 1], [1, 0]], columns=('x1', 'x2'))
data_set = PlotlyDataSet(
filepath='scatter_plot.json',
plotly_args={
'type': 'scatter',
'fig': {'x': 'x1', 'y': 'x2'},
}
)
data_set.save(df_data)
reloaded = data_set.load()
assert px.scatter(df_data, x='x1', y='x2') == reloaded