How To Calculate Unit Vector In Python