switch from graph_objects to px

Author: nicolaskruchten

Examples.ipynb

@@ -18,6 +18,7 @@
     "\n",
     "from plotly import figure_factory\n",
     "from plotly import graph_objects\n",
+    "import plotly.express as px\n",
     "import plotly.io as pio\n",
     "from IPython.core.magic import Magics, magics_class, cell_magic\n",
     "\n",
@@ -49,6 +50,11 @@
     "templated_fig = pio.to_templated(fig)\n",
     "pio.templates['my_template'] = templated_fig.layout.template\n",
     "pio.templates.default = 'my_template'\n",
+    "px.defaults.width = 100\n",
+    "px.defaults.height = 100\n",
+    "pio.renderers.default = \"png\"\n",
+    "pio.renderers[\"png\"].width = 900\n",
+    "pio.renderers[\"png\"].height = 900\n",
     "\n",
     "alt.renderers.enable('png', webdriver='firefox')"
    ]
@@ -286,17 +292,8 @@
    },
    "outputs": [],
    "source": [
-    "mpgGrouped = mpg.groupby('manufacturer').size()\n",
-    "fig = graph_objects.Figure(layout={'title' : 'Number of Cars by Make'})\n",
-    "bar = graph_objects.Bar({\n",
-    "        'type' : 'bar',\n",
-    "        'x' : mpgGrouped.values.tolist(),\n",
-    "        'y' : mpgGrouped.index.tolist(),\n",
-    "        'orientation' : 'h'\n",
-    "        \n",
-    "    })\n",
-    "fig.add_trace(bar)\n",
-    "Image(fig.to_image(format=\"png\", width=900, height=900))"
+    "px.histogram(mpg, y=\"manufacturer\", \n",
+    "             title='Number of Cars by Make')"
    ]
   },
   {
@@ -410,13 +407,7 @@
    },
    "outputs": [],
    "source": [
-    "fig = graph_objects.Figure()\n",
-    "hist = graph_objects.Histogram({\n",
-    "        'type' : 'histogram',\n",
-    "        'x' : mpg['cty'],\n",
-    "})\n",
-    "fig.add_trace(hist)\n",
-    "Image(fig.to_image(format=\"png\", width=900, height=900))"
+    "px.histogram(mpg, x=\"cty\")"
    ]
   },
   {
@@ -547,23 +538,12 @@
    },
    "outputs": [],
    "source": [
-    "fig = graph_objects.Figure(layout={\n",
-    "    'title' : 'Engine Displacement in Liters vs Highway MPG',\n",
-    "    'xaxis' : {\n",
-    "        'title' : 'Engine Displacement in Liters'\n",
-    "    },\n",
-    "    'yaxis' : {\n",
-    "        'title' : 'Highway MPG'\n",
-    "    }\n",
-    "})\n",
-    "scatter = graph_objects.Scatter({\n",
-    "    'type' : 'scatter',\n",
-    "    'mode' : 'markers',\n",
-    "    'x' : mpg.displ,\n",
-    "    'y' : mpg.hwy        \n",
-    "})\n",
-    "fig.add_trace(scatter)\n",
-    "Image(fig.to_image(format=\"png\", width=900, height=900))"
+    "px.scatter(mpg, x=\"displ\", y=\"hwy\", \n",
+    "           title='Engine Displacement in Liters vs Highway MPG',\n",
+    "           labels=dict(\n",
+    "               displ='Engine Displacement in Liters', \n",
+    "               hwy='Highway MPG')\n",
+    "          )"
    ]
   },
   {
@@ -841,27 +821,12 @@
    },
    "outputs": [],
    "source": [
-    "traces = []\n",
-    "for cls in mpg[\"class\"].unique():\n",
-    "    traces.append(\n",
-    "        graph_objects.Scatter(\n",
-    "            {\n",
-    "                \"mode\": \"markers\",\n",
-    "                \"x\": mpg.displ[mpg[\"class\"] == cls],\n",
-    "                \"y\": mpg.hwy[mpg[\"class\"] == cls],\n",
-    "                \"name\": cls,\n",
-    "            }\n",
-    "        )\n",
-    "    )\n",
-    "fig = graph_objects.Figure(\n",
-    "    layout={\n",
-    "        \"title\": \"Engine Displacement in Liters vs Highway MPG\",\n",
-    "        \"xaxis\": {\"title\": \"Engine Displacement in Liters\",},\n",
-    "        \"yaxis\": {\"title\": \"Highway MPG\"},\n",
-    "    },\n",
-    "    data=traces,\n",
-    ")\n",
-    "Image(fig.to_image(format=\"png\", width=900, height=900))"
+    "px.scatter(mpg, x=\"displ\", y=\"hwy\", color=\"class\", \n",
+    "           title='Engine Displacement in Liters vs Highway MPG',\n",
+    "           labels=dict(\n",
+    "               displ='Engine Displacement in Liters', \n",
+    "               hwy='Highway MPG')\n",
+    "          )"
    ]
   },
   {
@@ -976,29 +941,10 @@
    },
    "outputs": [],
    "source": [
-    "traces = [\n",
-    "    graph_objects.Scatter(\n",
-    "        {\n",
-    "            \"mode\": \"markers\",\n",
-    "            \"x\": mpg.cty,\n",
-    "            \"y\": mpg.hwy,\n",
-    "            \"marker\": {\"size\": mpg.cyl, \"color\": \"rgba(54,54,54,0.5)\"},\n",
-    "            \"name\": cls,\n",
-    "        }\n",
-    "    )\n",
-    "]\n",
-    "\n",
-    "fig = graph_objects.Figure(\n",
-    "    **{\n",
-    "        \"data\": traces,\n",
-    "        \"layout\": {\n",
-    "            \"title\": \"Engine Displacement in Liters vs Highway MPG\",\n",
-    "            \"xaxis\": {\"title\": \"Engine Displacement in Liters\",},\n",
-    "            \"yaxis\": {\"title\": \"Highway MPG\"},\n",
-    "        },\n",
-    "    }\n",
-    ")\n",
-    "Image(fig.to_image(format=\"png\", width=900, height=900))"
+    "px.scatter(mpg, x=\"cty\", y=\"hwy\", size=\"cyl\", size_max=10,\n",
+    "           title='City MPG vs Highway MPG',\n",
+    "           labels=dict(cty='City MPG', hwy='Highway MPG')\n",
+    "          )"
    ]
   },
   {
@@ -1099,8 +1045,7 @@
    },
    "outputs": [],
    "source": [
-    "fig = figure_factory.create_facet_grid(df=mpg, x=\"displ\", y=\"cty\", facet_col=\"class\")\n",
-    "Image(fig.to_image(format=\"png\", width=900, height=900))"
+    "px.scatter(mpg, x=\"displ\", y=\"hwy\", facet_col=\"class\", facet_col_wrap=4)"
    ]
   },
   {
@@ -1205,14 +1150,9 @@
    },
    "outputs": [],
    "source": [
-    "fig = figure_factory.create_facet_grid(\n",
-    "    df=mpg, \n",
-    "    x=\"displ\", \n",
-    "    y=\"cty\", \n",
-    "    facet_col=\"cyl\", \n",
-    "    facet_row=\"drv\"\n",
-    ")\n",
-    "Image(fig.to_image(format=\"png\", width=900, height=900))"
+    "px.scatter(mpg, x=\"displ\", y=\"hwy\", \n",
+    "           facet_col=\"cyl\", facet_row=\"drv\",\n",
+    "           category_orders=dict(cyl=[4,5,6,8]))"
    ]
   },
   {
@@ -1450,24 +1390,11 @@
    },
    "outputs": [],
    "source": [
-    "traces = []\n",
-    "newDiamond = diamonds.groupby(['cut','clarity']).size().unstack()\n",
-    "for c in newDiamond.columns:\n",
-    "    traces.append(graph_objects.Bar({\n",
-    "        'x' : newDiamond.index,\n",
-    "        'y' : newDiamond[c],\n",
-    "        'name' : c\n",
-    "    }))\n",
-    "fig = graph_objects.Figure(**{\n",
-    "    'data' : traces,\n",
-    "    'layout' : {\n",
-    "        'barmode' : 'stack',\n",
-    "        'xaxis' : {\n",
-    "            'title' : 'cut'\n",
-    "        },        \n",
-    "    }\n",
-    "})\n",
-    "Image(fig.to_image(format=\"png\", width=900, height=900))"
+    "px.histogram(diamonds, x=\"cut\", color=\"clarity\",\n",
+    "             category_orders=dict(cut=[\n",
+    "                 \"Fair\", \"Good\",  \"Very Good\", \n",
+    "                 \"Premium\", \"Ideal\"])\n",
+    "            )"
    ]
   },
   {
@@ -1558,24 +1485,11 @@
    },
    "outputs": [],
    "source": [
-    "traces = []\n",
-    "newDiamond = diamonds.groupby(['cut','clarity']).size().unstack()\n",
-    "for c in newDiamond.columns:\n",
-    "    traces.append(graph_objects.Bar({\n",
-    "        'x' : newDiamond.index,\n",
-    "        'y' : newDiamond[c],\n",
-    "        'name' : c\n",
-    "    }))\n",
-    "fig = graph_objects.Figure(**{\n",
-    "    'data' : traces,\n",
-    "    'layout' : {\n",
-    "        'barmode' : 'group',\n",
-    "        'xaxis' : {\n",
-    "            'title' : 'cut'\n",
-    "        },        \n",
-    "    }\n",
-    "})\n",
-    "Image(fig.to_image(format=\"png\", width=900, height=900))"
+    "px.histogram(diamonds, x=\"cut\", color=\"clarity\", barmode=\"group\",\n",
+    "             category_orders=dict(cut=[\n",
+    "                 \"Fair\", \"Good\",  \"Very Good\", \n",
+    "                 \"Premium\", \"Ideal\"])\n",
+    "            )"
    ]
   },
   {
@@ -1773,14 +1687,7 @@
    },
    "outputs": [],
    "source": [
-    "fig = graph_objects.Figure(layout={'xaxis' : { 'title' : 'date'}})\n",
-    "scatter = graph_objects.Scatter({\n",
-    "    'mode' :'lines',\n",
-    "    'x' : ts.date,\n",
-    "    'y' : ts.value\n",
-    "})\n",
-    "fig.add_trace(scatter)\n",
-    "Image(fig.to_image(format=\"png\", width=900, height=900))"
+    "px.line(ts, x=\"date\", y=\"value\")"
    ]
   },
   {
@@ -1818,7 +1725,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.0"
+   "version": "3.7.7"
   }
  },
  "nbformat": 4,

requirements.txt

@@ -15,7 +15,8 @@ chart-studio
 statsmodels==0.11.1
 rpy2
 psutil
-plotly
+plotly==4.10.0
+kaleido
 tzlocal
 simplegeneric
 altair