Question

我在tableform的文本文件中有三列数据。我使用np.genfromtxt将所有列读入matplotlib作为x，y，z。

我想创建一个颜色网格图，其中x和y是坐标，z代表颜色，我想人们将这样的图称为热图。

我的代码如下：

x = np.genfromtxt('mesh.txt', dtype=float, delimiter=' ', usecols = (0))
y = np.genfromtxt('mesh.txt', dtype=float, delimiter=' ', usecols = (1))
z = np.genfromtxt('mesh.txt', dtype=float, delimiter=' ', usecols = (2))

xmesh, ymesh = np.meshgrid(x,y)
diagram1.pcolormesh(xmesh,ymesh,z)

但是我收到以下错误消息：

line 7154, in pcolormesh
C = ma.ravel(C[0:Ny-1, 0:Nx-1]) # data point in each cell is value at
IndexError: too many indices

文本文件如下：

这是如何解决的。

Answer 1

在上面提供的示例数据中，可以轻松地重新整形x，y和z以获得2D阵列。下面的答案是针对那些正在寻找随机x，y和z数组的更广义答案的人。

import matplotlib.pyplot as plt
from matplotlib.mlab import griddata
import numpy

# use your x,y and z arrays here
x = numpy.random.randint(1,30, 50)
y = numpy.random.randint(1,30, 50)
z = numpy.random.randint(1,30, 50)

yy, xx = numpy.meshgrid(y,x)
zz = griddata(x,y,z,xx,yy, interp='linear')
plt.pcolor(zz)
#plt.contourf(xx,yy,zz) # if you want contour plot
#plt.imshow(zz)
plt.pcolorbar()
plt.show()

three_1D_arrays_2D_plot

Answer 2

我的猜测是，x，y和z将被读作相同长度的一维向量，假设为N.问题是当你创建xmesh和ymesh时，是N x N，你的z值也应该是。这只是N，这就是你收到错误的原因。

您的文件的布局是什么？我猜每一行都是你要创建网格的（x，y，z）。为此，您需要知道点如何作为网格排序（作为行主要或列主要）。知道了这一点后，您可以执行以下操作，而不是创建xmesh和ymesh：

N = np.sqrt(len(x)) # Only if squared, adjust accordingly
x = x.reshape((N, N))
y = y.reshape((N, N))
z = z.reshape((N, N))
pcolormesh(x, y, z)

在此之前，我会先这样做：

scatter(x, y, c=z)

它将为您提供网格的点，这是一个很好的起点。

Answer 3

我遇到了同样的问题，并同意Gustav Larsson建议的使用

scatter(x, y, c=z)

在我的特定情况下，我将散点的线宽设置为零：

scatter(x, y, c=z, linewidths=0)

当然，您可以使用其他装饰品，配色方案等，matplotlib.pyplot.scatter的参考将帮助您进一步。

Answer 4

似乎您正在将X和Y绘制为2D阵列，而Z仍然是1D阵列。尝试类似：

Znew=np.reshape(z,(len(xmesh[:,0]),len(xmesh[0,:])))
diagram1.pcolormesh(xmesh,ymesh,Znew)

更新： Tou有一个大小为4x4的X / Y网格：

x = np.genfromtxt('mesh.txt', dtype=float, delimiter=' ', usecols = (0))
y = np.genfromtxt('mesh.txt', dtype=float, delimiter=' ', usecols = (1))
z = np.genfromtxt('mesh.txt', dtype=float, delimiter=' ', usecols = (2))

按照@Gustav Larsson和我自己的建议重塑阵列：

Xnew=np.reshape(x,(4,4)) 
Xnew=np.reshape(y,(4,4))
Znew=np.reshape(z,(4,4))

这使您可以使用pcolormesh绘制三个4x4阵列：

diagram1.pcolormesh(Xnew,Ynew,Znew)

来自x，y，z数据的matplotlib pcolormesh图

4 个答案: