Python codes for correlation matrix and correlation calculation