refactoring

This commit is contained in:
frankknoll
2022-03-02 10:00:07 +01:00
parent 3bf5d4b41f
commit ef0544697a
2 changed files with 6 additions and 10 deletions

View File

@@ -72,16 +72,11 @@
" self.kreise = kreise\n",
"\n",
" def addKreisAndBundeslandAndEinwohnerzahlColumns(self, dataFrame):\n",
" dataFrame['Kreis'] = dataFrame['gemeindeschluessel'].map(\n",
" lambda gemeindeschluessel: self.kreise.loc[gemeindeschluessel, 3])\n",
"\n",
" dataFrame['Bundesland'] = dataFrame['bundesland'].map(\n",
" lambda bundesland: self.kreise.loc[bundesland, '2'])\n",
"\n",
" dataFrame['Einwohnerzahl'] = dataFrame['gemeindeschluessel'].map(\n",
" lambda gemeindeschluessel: int(self.kreise.loc[gemeindeschluessel, 6]))\n",
"\n",
" return 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['Bundesland'] = dataFrame['bundesland'].map(lambda bundesland: self.kreise.loc[bundesland, '2'])\n",
" return dataFrame"
]
},
{