diff --git a/src/intensivstationen/Intensivstationen.ipynb b/src/intensivstationen/Intensivstationen.ipynb index 3b3ddb2b9b9..e50b54af2b7 100644 --- a/src/intensivstationen/Intensivstationen.ipynb +++ b/src/intensivstationen/Intensivstationen.ipynb @@ -21,7 +21,13 @@ "metadata": {}, "outputs": [], "source": [ - "kreise = pd.read_excel('04-kreise.xlsx', sheet_name = 1, header = 5, index_col = 0)\n", + "kreise = pd.read_excel(\n", + " '04-kreise.xlsx',\n", + " sheet_name = 'Kreisfreie Städte u. Landkreise',\n", + " header = 5,\n", + " index_col = 0)\n", + "kreise = kreise.rename(columns = {'2': 'Bundesland', 3: 'Kreis', 6: 'Einwohnerzahl'})[['Bundesland', 'Kreis', 'Einwohnerzahl']]\n", + "kreise.index.set_names(\"Key\", inplace = True)\n", "kreise" ] }, @@ -73,8 +79,8 @@ "\n", " def addKreisAndBundeslandAndEinwohnerzahlColumns(self, dataFrame):\n", " dataFrame_kreise = pd.merge(dataFrame, self.kreise, how = 'left', left_on = 'gemeindeschluessel', right_index = True)\n", - " dataFrame['Kreis'] = dataFrame_kreise[3]\n", - " dataFrame['Einwohnerzahl'] = dataFrame_kreise[6]\n", + " dataFrame['Kreis'] = dataFrame_kreise['Kreis']\n", + " dataFrame['Einwohnerzahl'] = dataFrame_kreise['Einwohnerzahl']\n", " return self._addBundeslandColumn(dataFrame)\n", " \n", " def _addBundeslandColumn(self, dataFrame):\n", @@ -86,7 +92,7 @@ " right_index = True)\n", "\n", " def _createBundeslandByKeyTable(self):\n", - " bundeslandByKeyTable = self.kreise.reset_index().rename(columns = {1: 'Key', '2': 'Bundesland'})[['Key', 'Bundesland']]\n", + " bundeslandByKeyTable = self.kreise.reset_index()[['Key', 'Bundesland']]\n", " return bundeslandByKeyTable[bundeslandByKeyTable['Key'].str.len() == 2].set_index('Key')\n" ] },