Matplotlib imshow()cmap的对象太多了

时间:2013-08-28 01:56:32

标签: python matplotlib

我正在尝试创建一个2D高斯函数的简单imshow()图(matplotlib v.1.2.1):

import matplotlib.pyplot as plt
import numpy as np
from pylab import *

def gaussian(x,y,stdx,stdy):
    return 1.0/(2*np.pi*stdx*stdy) * (np.exp(-0.5*(x**2/stdx**2 + y**2/stdy**2)))

coords = np.linspace(-1,1,100)
X,Y = np.meshgrid(coords,coords)
std_list = np.linspace(1,2,20)
output = [gaussian(X,Y,std_list[i],std_list[i]) for i in range(len(std_list))]

for i in range(len(output)):
    plt.imshow(X,Y,np.array(output[i]),cmap='bone')
    plt.show()

我收到以下错误:

Traceback (most recent call last):
  File "blur.py", line 14, in <module>
    plt.imshow(X,Y,np.array(output[i]),cmap='bone')
TypeError: imshow() got multiple values for keyword argument 'cmap'

事实上,为了确保我没有疯狂,我完全取出了cmap参数,现在我收到以下错误:

Traceback (most recent call last):
  File "blur.py", line 14, in <module>
    plt.imshow(X,Y,np.array(output[i]))
  File "/home/username/anaconda/lib/python2.7/site-packages/matplotlib/pyplot.py", line     2737, in imshow
    imlim=imlim, resample=resample, url=url, **kwargs)
  File "/home/username/anaconda/lib/python2.7/site-packages/matplotlib/axes.py", line 7098, in imshow
    if norm is not None: assert(isinstance(norm, mcolors.Normalize))
AssertionError

我已经确定imshow()的参数都具有相同的形状,所以我不完全确定我做错了什么。这可能是个错误吗?

1 个答案:

答案 0 :(得分:6)

imshow不会将xyz作为输入。 (pcolorpcolormesh会这样做。

使用pcolormesh(x, y, z),或使用extent kwarg进行展示。

e.g。

plt.imshow(Z, extent=[X.min(), X.max(), Y.min(), Y.max()],
           cmap='bone')

plt.pcolormesh(X, Y, Z, cmap='bone')

正在发生的事情是imshow期待

imshow(X, cmap=None, norm=None, aspect=None, interpolation=None,
         alpha=None, vmin=None, vmax=None, origin=None, extent=None,
         **kwargs)

请注意,第二个参数是cmap,这解释了为什么在传递额外的cmap kwarg时出现错误的原因。

希望能让事情变得清晰!祝你好运!