https://llego.dev/posts/pandas-adding-columns-assign-python/