Python 資料相關性分析

2021-08-19 08:14:14 字數 4733 閱讀 5905

python方差協方差相關係數離散度pandasnumpy

接下來,我們將使用 anaconda 的 ipython 來演示如何使用 python 資料相關性分析,我所使用的 python 版本為 3.6.2 。

首先,我們將會建立兩個陣列,陣列內含有 20 個資料,均為 [0, 100] 區間內隨機生成。

a =

[random.randint(0,

100)

for a in

range(20

)]b =[random.randint(0,

100)

for a in

range(20

)]print

(a)>>[35

,2,75

,72,55

,77,69

,83,3

,46,31

,91,72

,12,15

,20,39

,18,57

,49]print

(b)>>[25

,24,72

,91,27

,44,85

,21,0

,64,44

,31,6

,91,1

,61,5

,39,24

,43]

在一般情況下,我們通過實驗或者調查統計獲取的資料很大一部分都屬於離散隨機變數,那麼這裡的期望我們也可以簡單的理解為平均數,那麼既然是平均數,那麼我們就可以非常簡單編寫乙個計算離散變數的期望的函式了。

def

mean

(x):

return

sum(x)

/len

(x)mean(a)

>>

46.05

mean(b)

>>

39.9

接下來,我們需要計算的是資料的離散程度,在統計上,我們通常會使用方差和標準差來描述。

方差和期望一樣,對於連續和離散的隨機變數有著不同的定義,具體定義如下:

與期望類似,這裡我們一般只考慮離散變數的方差。還有一點值得注意,我們上面的離散變數方差公式,最後是除以n,但實際上,我們計算樣本方差的時候一般會使用n-1,具體原因可以參考知乎 《為什麼樣本方差(sample variance)的分母是 n-1?》。

而標準差,就是方差的平方根。那麼,我們也可以像上面計算期望一樣,給方差和標準差編寫函式。

# 計算每一項資料與均值的差

defde_mean

(x):

x_bar = mean(x)

return

[x_i - x_bar for x_i in x]

# 輔助計算函式 dot product 、sum_of_squares

defdot

(v, w)

:return

sum(v_i * w_i for v_i, w_i in

zip(v, w)

)def

sum_of_squares

(v):

return dot(v, v)

# 方差

defvariance

(x):

n =len(x)

deviations = de_mean(x)

return sum_of_squares(deviations)

/(n -1)

# 標準差

import math

defstandard_deviation

(x):

return math.sqrt(variance(x)

)variance(a)

>>

791.8394736842105

varance(b)

>>

850.5157894736841

接下來,我們進入正題,我們開始計算兩組資料的相關性。我們一般採用相關係數來描述兩組資料的相關性,而相關係數則是由協方差除以兩個變數的標準差而得,相關係數的取值會在 [-1, 1] 之間,-1 表示完全負相關,1 表示完全相關。接下來,我們看一下協方差和相關係數的定義:

同樣的,我們根據上述的公式編寫函式。

# 協方差

defcovariance

(x, y)

: n =

len(x)

return dot(de_mean(x)

, de_mean(y))/

(n -1)

# 相關係數

defcorrelation

(x, y)

: stdev_x = standard_deviation(x)

stdev_y = standard_deviation(y)

if stdev_x >

0and stdev_y >0:

return covariance(x, y)

/ stdev_x / stdev_y

else

:return

0covariance(a, b)

>>

150.95263157894735

correlation(a, b)

>>

0.18394200852440826

根據上面的結果,相關係數為 0.18,可以推斷這兩組隨機數有弱正相關。當然,我們知道,這兩組資料都是使用 random 函式隨機生成出來的,其實並沒有什麼相關性,這也是在資料處理中,需要特別留意的乙個地方,統計的方法可以給我們乙個定量的數值可供分析,但實際的分析也需要結合實際以及更多的情況綜合考慮。

一般我們日常工作,都不會像上面一樣把什麼期望、方差、協方差一類的函式都重新寫一遍,上面的**只是讓我們對這些計算更加熟悉。我們通常情況下會使用 numpy 一類封裝好的函式,以下將演示一下如何使用 numpy 計算協方差。

import numpy as np

# 先構造乙個矩陣

ab = np.array(

[a, b]

)# 計算協方差矩陣

np.cov(ab)

>> array([[

791.83947368

,150.95263158],

[150.95263158

,850.51578947]]

)

這裡我們可以看到,這裡使用np.cov函式,輸出的結果是乙個矩陣,這就是協方差矩陣。協方差矩陣資料的看法也不難,我們可以以上面的結果為例,矩陣1行1列,表示的是 a 資料的方差,這和我們上面的計算結果一致,然後1行2列和2行1列分別是 a b 以及 b a 的協方差,所以他們的值是一樣的,然後最後2行2列就是 b 資料的方差。

接下來,我們繼續使用 numpy 計算相關係數

np.corrcoef(ab)

>> array([[

1.,0.18394201],

[0.18394201,1

.]])

計算相關係數,我們使用 numpy 的corrcoef函式,這裡的輸出也是乙個矩陣,這個矩陣資料的含義同上面的協方差類似,我們可以看到,這裡我們的相關係數是 0.18 ,和我們上面自己編寫的函式計算的結果一致。

除了使用 numpy,我們比較常用的 python 資料處理庫還有 pandas,很多金融資料分析的框架都會使用 pandas 庫,以下將演示如何使用 pandas 庫計算協方差和相關係數。

import pandas as pd

# 使用 dataframe 作為資料結構,為方便計算,我們會將 ab 矩陣轉置

dfab = pd.dataframe(ab.t, columns=

['a'

,'b'])

# a b 協方差

dfab.a.cov(dfab.b)

>>

150.95263157894738

# a b 相關係數

dfab.a.corr(dfab.b)

>>

0.18394200852440828

dfab

>> a b035

251224

275723

7291455

2757744669

857832183

094664

1031

4411

9131

1272613

1291

1415115

2061

1639517

1839

1857

2419

4943

可以看到,和 numpy 相比,pandas 對於有多組資料的協方差、相關係數的計算比 numpy 更為簡便、清晰,我們可以指定計算具體的兩組資料的協方差、相關係數,這樣就不需要再分析結果的協方差矩陣了。

本文通過建立兩組隨機的陣列,然後通過參考定義公式編寫函式,再到使用 numpy 以及 pandas 進行協方差、相關係數的計算。到這裡我們應該已經了解了資料相關性分析的原理,以及簡單的具體實踐使用方法,日後在工作中遇到需要做資料相關性分析的時候,就可以派上用場了。

《資料科學入門.格魯斯 (joel grus).人民郵電出版社》

Python 相關性分析

資料來源示例 餐飲銷量資料相關性分析 import pandas as pd catering sale data catering sale all.xls 餐飲資料,含有其他屬性 data pd.read excel catering sale,index col u 日期 讀取資料,指定 日期...

spearman相關性分析 相關性分析

r語言常用函式 cor 預設結果為矩陣 cor mydat,use method use 缺失值的處理,method 處理方法 cor x,y 可以計算非方形矩陣,x y分別為2個矩陣,相同的行數 cor.test x,y,alternative method x y為檢驗相關性的變數 librar...

資料相關性分析

源於 python資料分析與挖掘實戰 coding utf 8 import pandas as pd catering sale data catering sale all.xls data pd.read excel catering sale,index col u 日期 data.corr...