/
correlation.py
38 lines (34 loc) · 916 Bytes
/
correlation.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import numpy as np
import matplotlib.pyplot as plt
import multiprocessing as mp
import datetime as dt
import matplotlib.dates as mdates
from scipy import stats
import pandas as pd
import requests
import json
import math
import csv
def csv_read(file):
csvfile = open(file, 'r')
reader = csv.reader(csvfile)
A = []
for row in reader:
A.append(float(row[1]))
return A
def csv_write(file, A):
csvfile = open(file, 'w')
writer = csv.writer(csvfile)
writer.writerow(A)
csvfile.close()
def main():
while True:
file1 = input("Name of first csv file: ")
file2 = input("Name of second csv file: ")
data1 = csv_read(file1)
data2 = csv_read(file2)
corr = [data1, data2]
print("correlation matrix: ", np.corrcoef(corr))
print("chi-square p-value: ", stats.chi2_contingency(corr)[1])
if __name__ == '__main__':
main()