Хотя ответ Иканобори - путь сюда, я подумал, что покажу другую простую альтернативу.
import numpy as np
import matplotlib.pyplot as plt
import bpython
bpython.embed(locals_=locals())
Это запустит REPL bpython и загрузит локальные переменные и другие символы. Это было бы полезно, если вы хотите иметь более одной настраиваемой оболочки.