Colorbar for density plots

asked 2012-10-23 20:57:25 +0200

MelvinSmith
21 ●1 ●3

Hi!

I would like to know if there is a way to get a colorbar for a density plot. For contour plots this works fine with the colorbar=True option, but for density plots there doesn't seem to be such an option. Can anyone help?

Greetings, Melvin

edit retag flag offensive close merge delete

add a comment

2 Answers

Sort by » oldest newest most voted

answered 2012-10-23 23:20:24 +0200

Shashank

1917 ●30 ●51 ●83

I don't know the answer in sage but it is easy in matplotlib

import matplotlib
import matplotlib.cm as cm
import matplotlib.pyplot as plt
import numpy as np


fig = plt.figure()
array=np.zeros((100,100))
for i in range(100):
    for j in range(100):
        array[i][j]=sin(0.1*i)*cos(0.1*j)
#norm=matplotlib.colors.Normalize(0.995,1.0)
im1 = plt.imshow(array, cmap=cm.hot, origin='lower',clip_on=True,filternorm=1)
#im1.set_norm(norm)
plt.colorbar()
plt.savefig('test.png')