plot with python