I\'m learning python and trying to adapt a notebook someone posted on Kaggle to my current project. Unfortunately, I keep getting a "Length Mismatch: Expected Axis has