Keep environment variable while using sudo

$ export A="B"

# Use -E to keep the above environment variable
$ sudo -E python a.py