I found another solution that may be better,
use the Python Data Science template instead of a normal Python one
It may take a minute or two to load the nix environment on the first run. (55 seconds for packager, and then after another 40 seconds the output window appears and then after about another 15 seconds it actually shows the plot)
