Построение 2D-карты тепла с помощью Matplotlib

Question

Используя Matplotlib, я хочу построить двумерную карту тепла. Мои данные представляют собой массив n-by-n Numpy, каждый из которых имеет значение от 0 до 1. Итак, для элемента (i, j) этого массива я хочу построить квадрат в координате (i, j) в моей тепловая карта, цвет которой пропорционален значению элемента в массиве.

Как я могу это сделать?

Answer 1

Функция imshow() с параметрами interpolation='nearest' и cmap='hot' должна делать то, что вы хотите.

import matplotlib.pyplot as plt 
import numpy as np 

a = np.random.random((16, 16)) 
plt.imshow(a, cmap='hot', interpolation='nearest') 
plt.show() 

Answer 2

Вот как это сделать из CSV:

import numpy as np 
import matplotlib.pyplot as plt 
from scipy.interpolate import griddata 

# Load data from CSV 
dat = np.genfromtxt('dat.xyz', delimiter=' ',skip_header=0) 
X_dat = dat[:,0] 
Y_dat = dat[:,1] 
Z_dat = dat[:,2] 

# Convert from pandas dataframes to numpy arrays 
X, Y, Z, = np.array([]), np.array([]), np.array([]) 
for i in range(len(X_dat)): 
     X = np.append(X,X_dat[i]) 
     Y = np.append(Y,Y_dat[i]) 
     Z = np.append(Z,Z_dat[i]) 

# create x-y points to be used in heatmap 
xi = np.linspace(X.min(),X.max(),1000) 
yi = np.linspace(Y.min(),Y.max(),1000) 

# Z is a matrix of x-y values 
zi = griddata((X, Y), Z, (xi[None,:], yi[:,None]), method='cubic') 

# I control the range of my colorbar by removing data 
# outside of my range of interest 
zmin = 3 
zmax = 12 
zi[(zi<zmin) | (zi>zmax)] = None 

# Create the contour plot 
CS = plt.contourf(xi, yi, zi, 15, cmap=plt.cm.rainbow, 
        vmax=zmax, vmin=zmin) 
plt.colorbar() 
plt.show() 

где dat.xyz находится в форме

x1 y1 z1 
x2 y2 z2 
...