Если вы имеете в виду корреляцию, как в R ^ 2 Excel, вы можете использовать что-то вроде этого (уже сделал это для моей работы):
def correlation(Measure, Fit):
"""Calculates the correlation coefficient R^2 between the two sets
of Y data provided. Logically, in order for the result to have a sense
you want both Y arrays to have been created from the same X array."""
Mean = np.mean(Measure)
s1 = 0
s2 = 0
Size = np.size(Measure) # identical to np.size(Fit)
for i in range(0, Size):
s1 += (Measure[i] - Fit[i]) ** 2
s2 += (Measure[i] - Mean) ** 2
Rsquare = 1 - s1/s2
return Rsquare
И я удалил их для удобства чтения, но вы можетеокружите это различными мерами предосторожности и сообщениями об ошибках, например, когда оба массива не имеют одинаковый размер или содержат NAN.
Редактировать: используемая формула взята из статьи о коэффициенте определения в Википедии.