pca (3 dimensions) with python