.zshrc

python_run() {
  $@ 2>.pythonrunerr
  if [ $? -ne 0 ]; then
    sed -nri -e '/^\s*File/N' -e 's/^\s*?File "(.*?)", line (.*?),.*\n(.*)$/\1:\2:\3/p' .pythonrunerr
    vim -q .pythonrunerr -c cope
  else
    cat .pythonrunerr >&2
  fi
  rm .pythonrunerr
}

example

$ python_run ./test.py

test.py

#!/usr/bin/env python

import os, sys

def a():
    b()

def b():
    c()

def c():
    d()

def d():
    e() #NameError

def main(env, argv):
    a()
    return 0


if __name__ == '__main__':
    exit(main(os.environ, sys.argv))