refactoring

This commit is contained in:
frankknoll
2022-03-02 11:17:04 +01:00
parent 480b1dcca0
commit e0e749f552

View File

@@ -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"
]
},