pearson correlation python example