https://itsmycode.com/how-to-fix-keyerror-in-pandas/