dimensionality reduction python code