2013-08-28 108 views
2

我试图创建一个简单的imshow()绘制2D高斯函数(matplotlib v.1.2.1)太多的对象:Matplotlib imshow()为CMAP

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()的参数都具有相同的形状,所以我不完全确定我做错了什么。这可能是一个错误?

回答

6

imshow不需要x,y,z作为输入。 (然而,pcolorpcolormesh)。

可以使用pcolormesh(x, y, z)或使用extent kwarg来imshow。

例如

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克瓦格。

希望能够澄清一些事情!祝你好运!

+0

感谢您的快速响应!我总是忘记哪些函数使用范围,哪些可以用np.ndarrays作为输入。 – astromax