Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Scatter Plot Feature #24

Merged
merged 28 commits into from Sep 30, 2021
Merged

Scatter Plot Feature #24

merged 28 commits into from Sep 30, 2021

Conversation

teubert
Copy link
Collaborator

@teubert teubert commented Sep 22, 2021

Add new feature to produce a scatter plot of states. Also, added to basic example and tutorial.

Can be called directly from the result of the state estimator (UKF or PF). Like so:

states.plot_scatter() # Simple scatter chart

Also allows you to overlay multiple states on one chart:

fig = initial_states.plot_scatter() # Simple scatter chart
future_states.plot_scatter(fig=fig) # Add future states to same chart

Everything that's configurable from the scatter function in matplotlib is configurable using kwargs, e.g.,

fig = initial_states.plot_scatter(c='b') # Blue points

Also, you can specify a subset of the keys to plot. E.g.,

fig = initial_states.plot_scatter(keys=['state1', 'state2']) # Only plot state 1 and state 2

Here's what it looks like:
Screen Shot 2021-09-21 at 5 50 50 PM

@teubert teubert added the enhancement New feature or request label Sep 22, 2021
@teubert teubert linked an issue Sep 22, 2021 that may be closed by this pull request
@teubert teubert requested review from matteocorbetta and removed request for matteocorbetta September 22, 2021 18:46
@teubert teubert merged commit 6b70c1a into dev Sep 30, 2021
@teubert teubert deleted the feature/scatter branch September 30, 2021 19:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

State Scatter Plots
2 participants