{ "cells": [ { "cell_type": "code", "execution_count": 1, "metadata": {}, "outputs": [ { "data": { "text/html": [ " \n", " " ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "from functools import reduce\n", "\n", "import pandas as pd\n", "import plotly.graph_objects as go\n", "from plotly import express as px\n", "from plotly.colors import n_colors\n", "from plotly.subplots import make_subplots\n", "from plotly.offline import init_notebook_mode\n", "\n", "init_notebook_mode()\n", "pd.options.plotting.backend = \"plotly\"" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "This is my attempt on analyzing the data collected by [fivethirtyeight](https://fivethirtyeight.com/) when trying to determine which candy is the best.\n", "You can read more about the data set, how it was collected etc. [here](https://fivethirtyeight.com/videos/the-ultimate-halloween-candy-power-ranking/).\n", "The data itself is available [here](https://github.com/fivethirtyeight/data/blob/master/candy-power-ranking/candy-data.csv) (don't forget to have a look in the [README](https://github.com/fivethirtyeight/data/blob/master/candy-power-ranking/README.md))." ] }, { "cell_type": "code", "execution_count": 2, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", " | competitorname | \n", "chocolate | \n", "fruity | \n", "caramel | \n", "peanutyalmondy | \n", "nougat | \n", "crispedricewafer | \n", "hard | \n", "bar | \n", "pluribus | \n", "sugarpercent | \n", "pricepercent | \n", "winpercent | \n", "
---|---|---|---|---|---|---|---|---|---|---|---|---|---|
0 | \n", "100 Grand | \n", "1 | \n", "0 | \n", "1 | \n", "0 | \n", "0 | \n", "1 | \n", "0 | \n", "1 | \n", "0 | \n", "0.732 | \n", "0.860 | \n", "66.971725 | \n", "
1 | \n", "3 Musketeers | \n", "1 | \n", "0 | \n", "0 | \n", "0 | \n", "1 | \n", "0 | \n", "0 | \n", "1 | \n", "0 | \n", "0.604 | \n", "0.511 | \n", "67.602936 | \n", "
2 | \n", "One dime | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0.011 | \n", "0.116 | \n", "32.261086 | \n", "
3 | \n", "One quarter | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0.011 | \n", "0.511 | \n", "46.116505 | \n", "
4 | \n", "Air Heads | \n", "0 | \n", "1 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0.906 | \n", "0.511 | \n", "52.341465 | \n", "
... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "
80 | \n", "Twizzlers | \n", "0 | \n", "1 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0.220 | \n", "0.116 | \n", "45.466282 | \n", "
81 | \n", "Warheads | \n", "0 | \n", "1 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "1 | \n", "0 | \n", "0 | \n", "0.093 | \n", "0.116 | \n", "39.011898 | \n", "
82 | \n", "WelchÕs Fruit Snacks | \n", "0 | \n", "1 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "1 | \n", "0.313 | \n", "0.313 | \n", "44.375519 | \n", "
83 | \n", "WertherÕs Original Caramel | \n", "0 | \n", "0 | \n", "1 | \n", "0 | \n", "0 | \n", "0 | \n", "1 | \n", "0 | \n", "0 | \n", "0.186 | \n", "0.267 | \n", "41.904308 | \n", "
84 | \n", "Whoppers | \n", "1 | \n", "0 | \n", "0 | \n", "0 | \n", "0 | \n", "1 | \n", "0 | \n", "0 | \n", "1 | \n", "0.872 | \n", "0.848 | \n", "49.524113 | \n", "
85 rows × 13 columns
\n", "\n", " | Trait | \n", "Median (without) | \n", "Median (with) | \n", "Median change (pct) | \n", "Presence (pct) | \n", "
---|---|---|---|---|---|
8 | \n", "pluribus | \n", "52.341465 | \n", "45.866287 | \n", "-12.371028 | \n", "51.764706 | \n", "
9 | \n", "high_sugar | \n", "46.056166 | \n", "54.861111 | \n", "19.117842 | \n", "45.882353 | \n", "
1 | \n", "fruity | \n", "56.490501 | \n", "42.969034 | \n", "-23.935824 | \n", "44.705882 | \n", "
0 | \n", "chocolate | \n", "41.646933 | \n", "60.800701 | \n", "45.990827 | \n", "43.529412 | \n", "
7 | \n", "bar | \n", "43.732221 | \n", "60.800701 | \n", "39.029528 | \n", "24.705882 | \n", "
6 | \n", "hard | \n", "52.583706 | \n", "39.460556 | \n", "-24.956685 | \n", "17.647059 | \n", "
2 | \n", "caramel | \n", "46.411716 | \n", "60.164976 | \n", "29.633164 | \n", "16.470588 | \n", "
3 | \n", "peanutyalmondy | \n", "45.995827 | \n", "70.109714 | \n", "52.426251 | \n", "16.470588 | \n", "
4 | \n", "nougat | \n", "46.978288 | \n", "60.800701 | \n", "29.422980 | \n", "8.235294 | \n", "
5 | \n", "crispedricewafer | \n", "46.354157 | \n", "66.470680 | \n", "43.397453 | \n", "8.235294 | \n", "