Retrieve XY data from matplotlib figure [duplicate]

by

This works:

In [1]: import matplotlib.pyplot as plt

In [2]: plt.plot([1,2,3],[4,5,6])
Out[2]: [<matplotlib.lines.Line2D at 0x30b2b10>]

In [3]: ax = plt.gca() # get axis handle

In [4]: line = ax.lines[0] # get the first line, there might be more

In [5]: line.get_xdata()
Out[5]: array([1, 2, 3])

In [6]: line.get_ydata()
Out[6]: array([4, 5, 6])

In [7]: line.get_xydata()
Out[7]: 
array([[ 1.,  4.],
       [ 2.,  5.],
       [ 3.,  6.]])

I found these by digging around in the axis object. I could only find some minimal information about these functions, apperently you can give them a boolean flag to get either original or processed data, not sure what the means.

Edit: Joe Kington showed a slightly neater way to do this:

In [1]: import matplotlib.pyplot as plt

In [2]: lines = plt.plot([1,2,3],[4,5,6],[7,8],[9,10])

In [3]: lines[0].get_data()
Out[3]: (array([1, 2, 3]), array([4, 5, 6]))

In [4]: lines[1].get_data()
Out[4]: (array([7, 8]), array([ 9, 10]))

More Related Contents:

  1. Embedding a matplotlib figure inside a WxPython panel
  2. Adding value labels on a matplotlib bar chart
  3. setting y-axis limit in matplotlib
  4. Pandas Dataframe line plot display date on xaxis
  5. Matplotlib discrete colorbar
  6. Purpose of “%matplotlib inline”
  7. Hiding axis text in matplotlib plots
  8. How to plot and annotate a grouped bar chart
  9. Moving x-axis to the top of a plot in matplotlib
  10. How to display custom values on a bar plot
  11. matplotlib: how to prevent x-axis labels from overlapping
  12. Matplotlib issue on OS X (“ImportError: cannot import name _thread”)
  13. How to change figuresize using seaborn factorplot
  14. How to make a 4d plot using Python with matplotlib
  15. Passing a matplotlib figure to HTML (flask)
  16. How can I write unit tests against code that uses matplotlib?
  17. Color matplotlib plot_surface command with surface gradient
  18. Many plots in less time – python
  19. Stacked bar chart in Seaborn
  20. Improve subplot size/spacing with many subplots
  21. sklearn plot confusion matrix with labels
  22. How to plot multiple bars grouped
  23. How to draw axis in the middle of the figure?
  24. pandas – scatter plot with different color legend for each point
  25. iterating markers in plots
  26. matplotlib generic colormap from tab10
  27. Plot inline or a separate window using Matplotlib in Spyder IDE
  28. import matplotlib.pyplot hangs
  29. Is it possible to add hatches to each individual bar in seaborn.barplot?
  30. matplotlib taking time when being imported

Leave a Comment