Getting the numpy array describing an arbitrary matplolib/seaborn plot

Question

Is it possible to get the numpy array describing an arbitrary matplolib/seaborn plot without explicitly saving it as an image file and reading it?

For example, say I have the attached image which is a sns.kdeplot() of an array. Now, can I get the numpy array of this plot without first saving it to a file and reading it? Something like:

img = sns.kdeplot(arr)
img_arr = img.some_function() # Returns a numpy array describing the plot

Quang Hoang · Accepted Answer · 2019-10-25 20:00:27Z

2

fig.canvas.tostring_rgb() would be helpful:

fig, ax = plt.subplots()
sns.kdeplot(arr, ax=ax)

img_arr = np.fromstring(fig.canvas.tostring_rgb(), 
                        dtype=np.uint8,
                        sep='')

answered Oct 25, 2019 at 20:00

Quang Hoang

151k11 gold badges64 silver badges86 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Getting the numpy array describing an arbitrary matplolib/seaborn plot

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related