И добавьте, пожалуйста, комментарии сбоку, что означает каждое действие. И всё должно работать в Jupyter.