Using An Array In An If Statement In Python