Подтвердить что ты не робот

Как получить корреляцию двух векторов в python

В Matlab я использую

a=[1,4,6]
b=[1,2,3]
corr(a,b)

который возвращает .9934. Я пробовал numpy.correlate, но он возвращает что-то совершенно другое. Каков самый простой способ получить корреляцию двух векторов?

Ответ 1

Документы показывают, что numpy.correlate не то, что вы ищете:

numpy.correlate(a, v, mode='valid', old_behavior=False)[source]
  Cross-correlation of two 1-dimensional sequences.
  This function computes the correlation as generally defined in signal processing texts:
     z[k] = sum_n a[n] * conj(v[n+k])
  with a and v sequences being zero-padded where necessary and conj being the conjugate.

Вместо этого, как и другие комментарии, вы ищете коэффициент корреляции Пирсона. Чтобы сделать это с помощью scipy try:

from scipy.stats.stats import pearsonr   
a = [1,4,6]
b = [1,2,3]   
print pearsonr(a,b)

Это дает

(0.99339926779878274, 0.073186395040328034)

Вы также можете использовать numpy.corrcoef:

import numpy
print numpy.corrcoef(a,b)

Это дает:

[[ 1.          0.99339927]
 [ 0.99339927  1.        ]]