daviddoji/Kalman-and-Bayesian-Filters-in-Python
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

lots of pretty graphs

Browse Source
This commit is contained in:
Roger Labbe 2014-05-01 16:13:33 -05:00
parent d3800c5a4f
commit 20bd353b8a
2 changed files with 125 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

132
Multidimensional Kalman Filters.ipynb
View File

File diff suppressed because one or more lines are too long

15
gaussian.py
View File

@ -93,10 +93,11 @@ def plot_sigma_ellipse(ellipse,title=None):
plt.scatter(x,y,marker='+') # mark the center
if title is not None:
plt.title (title)
plt.show()
def plot_sigma_ellipses(ellipses,title=None,axis_equal=True,x_lim=None,y_lim=None):
""" plots the ellipse produced from sigma_ellipse."""
isct = plt.Circle((2,2),1,color='b',fill=False)
plt.figure().gca().add_patch(isct)
if x_lim is not None:
axis_equal = False
@ -120,8 +121,6 @@ def plot_sigma_ellipses(ellipses,title=None,axis_equal=True,x_lim=None,y_lim=Non
plt.scatter(x,y,marker='+') # mark the center
if title is not None:
plt.title (title)
plt.show()
def _to_array(x):
""" returns any of a scalar, matrix, or array as a 1D numpy array
@ -165,4 +164,14 @@ if __name__ == '__main__':
assert rv.pdf(1.2) == x2
assert abs(x2- x3) < 0.00000001
cov = np.array([\
[1,1],\
[1,1.1]])
ev = sigma_ellipses(cov, x=2, y=2, sigma=sigma)
plot_sigma_ellipses([e1, e2, e3, ev], axis_equal=True,x_lim=[0,4],y_lim=[0,15])
#isct = plt.Circle((2,2),1,color='b',fill=False)
#plt.figure().gca().add_artist(isct)
plt.show()
print "all tests passed"