user2086122
user2086122

Reputation: 65

correlation in statsmodel using python

could u please help me to Find correlation for these two lists importing stats-model in python.

  a=[1.0,2.0,3.0,2.0]
  b=[789.0,786.0,788.0,785.0]

using some built-in functions

Upvotes: 1

Views: 7941

Answers (1)

Vincent
Vincent

Reputation: 17715

>>> import numpy as np
>>> a = np.array([1.0,2.0,3.0,2.0])
>>> b = np.array([789.0,786.0,788.0,785.0])
>>> np.corrcoef(a,b)
array([[ 1.       , -0.2236068],
       [-0.2236068,  1.       ]])

Just use indexing to extract the right one:

np.corrcoef(a,b)[0,1]

Upvotes: 3

Related Questions