首页 > 解决方案 > 如何绘制均值为 0 的协方差误差椭圆?

问题描述

我正在尝试绘制不确定椭圆,它是与平均值相差一个标准差的点的椭圆。我正在使用此代码来执行此操作:https ://stackoverflow.com/a/25022642/7448860

但是,它画的是一条线,而不是椭圆。

我正在绘制 2 个随机变量 (x, y),因此协方差矩阵的大小为 2 x 2。要计算协方差矩阵,我使用的是卡尔曼滤波器算法,所以我不需要使用 np.cov()。

代码:

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.patches import Ellipse

def eigsorted(cov):
    vals, vecs = np.linalg.eigh(cov)
    order = vals.argsort()[::-1]

    return vals[order], vecs[:,order]

x = [0]
y = [0]
cov = [[0.25, 0.5 ], [0.5, 1.]]

nstd = 1
ax = plt.subplot(111)

vals, vecs = eigsorted(cov)

theta = np.degrees(np.arctan2(*vecs[:,0][::-1]))
w, h = 2 * nstd * np.sqrt(vals)
ell = Ellipse(xy=(np.mean(x), np.mean(y)),
              width=w, height=h,
              angle=theta, color='black')

ell.set_facecolor('none')
ax.add_artist(ell)
plt.scatter(x, y)
plt.show()

输出: 在此处输入图像描述

如何绘制距离均值一个标准差的椭圆?

我试过了:

x = [0,1,0,1]
y = [0,0,1,1]

但是,它绘制: 在此处输入图像描述

标签: pythonnumpymatplotlib

解决方案


推荐阅读