Python matplotlib:“文件”;TypeError:图像数据无法转换为浮点;看起来像是一个很好的矩阵
我知道其他人也看到了类似的错误(,),但我看不到任何解决方案对我有帮助 我试图用浮点数据填充numpy数组,并使用imshow绘制它。Y方向(几乎)的数据是厄米多项式和高斯包络,而X方向只是高斯包络Python matplotlib:“文件”;TypeError:图像数据无法转换为浮点;看起来像是一个很好的矩阵,python,numpy,matplotlib,Python,Numpy,Matplotlib,我知道其他人也看到了类似的错误(,),但我看不到任何解决方案对我有帮助 我试图用浮点数据填充numpy数组,并使用imshow绘制它。Y方向(几乎)的数据是厄米多项式和高斯包络,而X方向只是高斯包络 from __future__ import print_function import numpy as np import matplotlib.pyplot as plt ####First we set Ne Ne=25 ###Set up a mesh with size sqrt(N
from __future__ import print_function
import numpy as np
import matplotlib.pyplot as plt
####First we set Ne
Ne=25
###Set up a mesh with size sqrt(Ne) X sqrt(Ne)
sqrtNe=int(np.sqrt(Ne))
Ky=np.array(range(-sqrtNe,sqrtNe+1),dtype=float)
Kx=np.array(range(-sqrtNe,sqrtNe+1),dtype=float)
[KXmesh,KYmesh]=np.meshgrid(Kx,Ky,indexing='ij')
##X-direction is gussian envelope
AxMesh=np.exp(-(np.pi*KXmesh**2)/(4.0*Ne))
Nerror=21 ###This is where the error shows up
for n in range(Nerror,Ne):
##Y-direction is a polynomial of degree n ....
AyMesh=0.0
for i in range(n/2+1):
AyMesh+=(-1)**i*(np.sqrt(2*np.pi)*2*KYmesh)**(n-2*i)/(np.math.factorial(n-2*i)*np.math.factorial(i))
### .... times a gaussian envelope
AyMesh=AyMesh*np.exp(-np.pi*KYmesh**2)
AyMesh=AyMesh/np.max(np.abs(AyMesh))
WeightMesh=AyMesh*AxMesh
print("n:",n)
plt.figure()
####Error occurs here #####
plt.imshow(WeightMesh,interpolation='nearest')
plt.show(block=False)
当代码到达impow时,我会收到以下错误消息
Traceback (most recent call last):
File "FDOccupation_mimimal.py", line 30, in <module>
plt.imshow(WeightMesh,interpolation='nearest')
File "/usr/lib/python2.7/dist-packages/matplotlib/pyplot.py", line 3022, in imshow
**kwargs)
File "/usr/lib/python2.7/dist-packages/matplotlib/__init__.py", line 1814, in inner
return func(ax, *args, **kwargs)
File "/usr/lib/python2.7/dist-packages/matplotlib/axes/_axes.py", line 4947, in imshow
im.set_data(X)
File "/usr/lib/python2.7/dist-packages/matplotlib/image.py", line 449, in set_data
raise TypeError("Image data can not convert to float")
TypeError: Image data can not convert to float
简单地
AyMesh=KYmesh**n*np.exp(-np.pi*KYmesh**2)
AyMesh=AyMesh/np.max(np.abs(AyMesh))
问题消失了!?
有人知道这里发生了什么吗?对于大值,
np.math.factorial
返回一个long
而不是int
。具有long
值的数组属于数据类型object
,因为不能使用NumPy的类型存储。您可以通过以下方式重新转换最终结果:
WeightMesh=np.array(AyMesh*AxMesh, dtype=float)
要有一个合适的浮点数组
WeightMesh=np.array(AyMesh*AxMesh, dtype=float)