Tengo datos en un archivo de texto en forma de tabla con tres columnas. Uso np.genfromtxt para leer todas las columnas en matplotlib como x, y, z.

Quiero crear una trama de malla de color donde x e y son las coordenadas yz representa el color, creo que la gente se refiere a dicha trama como mapa de calor.

Mi código es el siguiente:

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)   

Pero me sale el siguiente mensaje de error:

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

El archivo de texto es el siguiente:

1 1 5
2 1 4
3 1 2
4 1 6
1 2 6
2 2 2
3 2 1
4 2 9
1 3 7
2 3 4
3 3 3
4 3 5
1 4 3
2 4 4
3 4 7
4 4 6

¿Cómo se soluciona esto?