我想知道是否有人可以给我一些有关如何在Python中计算协方差的技巧;我不想使用numpy中的任何内容。我只想学习如何手动执行此操作,并开始练习for循环。

基本上,我想计算以下的协方差:

X = [1,2]
Y = [1,2,3]
P = [[0.25,0.25,0.0], [0.0, 0.25, 0.25]]

Mean of X: 1.5
Mean of Y: 2


这些值取自:https://onlinecourses.science.psu.edu/stat414/node/109

其结果应为0.25。

我一直在嵌套的循环中遍历X,Y和P,但是不知道我可以使用其他方法来组合它。

我基本上想进行以下计算:

(1-1.5)(1-2)(0.25) + (1-1.5)(2-2)(0.25) +  ..... + (2-1.5)(3-2)(0.25)

最佳答案

要计算协方差,您需要类似下面的内容,它具有一个嵌套循环,遍历每个列表,并使用协方差公式累加协方差。

# let's get the mean of `X` (add all the vals in `X` and divide by
# the length
x_mean = float(sum(X)) / len(X)

# now, let's get the mean for `Y`
y_mean = float(sum(Y)) / len(Y)

# initialize the covariance to 0 so we can add it up
cov = 0

# we'll use a nested loop structure -- the outer loop can be through `Y`
# or `X`, it doesn't matter in this case
# we'll use python's `enumerate`, which lets us iterate through the `list`
# using a `tuple` that contains (the_current_index, the_current_element),
# or in `C`/`Java` terms, `(i, arr[i])`
for y_idx,y in enumerate(Y):
    for x_idx,x in enumerate(X):

        # the covariance is defined by the following equation
        # you don't need to loop through `P` -- the outer list
        # contains 2 elements, which is the size of `X`, and
        # the inner list contains 3 elements, which is the size of `Y`
        cov += (x - x_mean) * (y - y_mean) * P[x_idx][y_idx]

print cov # => 0.25

关于python - 用Python计算协方差,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/33797330/

10-15 22:53