Если вы делаете это в интерактивном режиме, лучше всего установить ipython и настроить параметры запуска для включения этого кода. Если вы намереваетесь включить его в сценарий, запускаемый из интерпретатора, то же самое относится и к нему, поскольку он будет иметь доступ к вашему пространству имен.
С другой стороны, автономный скрипт должен не включать это автоматически. В будущем вы или другой сопровождающий придете, и весь код должен быть очевиден и не зависеть от конкретной настройки машины. Лучше всего создать файл скелета для новых проектов, который включает в себя все основные функциональные возможности, которые вам нужны. Это, наряду с часто используемыми фрагментами, решит проблему.
Весь ваш код для запуска сценария будет в сценарии, и вам не придется каждый раз добавлять этот код.