plot.py: remove redundant numpy import