Tạo bản đồ nhiệt bằng cách sử dụng gấu trúc TimeSeries

Question

Tôi cần tạo bản đồ nhiệt MatplotLib (pcolormesh) bằng cột Thời gian Pandas DataFrame (df_all.ts) làm trục X của tôi.

Làm thế nào để chuyển đổi cột Pandas TimeSeries thành một thứ có thể dùng làm trục X trong hàm np.meshgrid (x, y) để tạo bản đồ nhiệt? Cách giải quyết là tạo Matplotlib drange bằng các tham số giống như trong cột gấu trúc, nhưng có cách đơn giản không?

x = pd.date_range(df_all.ts.min(),df_all.ts.max(),freq='H') 
xt = mdates.drange(df_all.ts.min(), df_all.ts.max(), dt.timedelta(hours=1)) 
y = arange(ylen) 
X,Y = np.meshgrid(xt, y) 

Answer 1

Tôi không biết những gì bạn có nghĩa là bởi bản đồ nhiệt cho một chuỗi thời gian, nhưng đối với một dataframe bạn có thể làm như sau:

import numpy as np 
import pandas as pd 
import matplotlib.pyplot as plt 

from itertools import product 
from string import ascii_uppercase 
from matplotlib import patheffects 

m, n = 4, 7 # 4 rows, 7 columns 
df = pd.DataFrame(np.random.randn(m, n), 
        columns=list(ascii_uppercase[:n]), 
        index=list(ascii_uppercase[-m:])) 


ax = plt.imshow(df, interpolation='nearest', cmap='Oranges').axes 

_ = ax.set_xticks(np.linspace(0, n-1, n)) 
_ = ax.set_xticklabels(df.columns) 
_ = ax.set_yticks(np.linspace(0, m-1, m)) 
_ = ax.set_yticklabels(df.index) 

ax.grid('off') 
ax.xaxis.tick_top() 

tùy chọn, in giá trị thực tế ở giữa mỗi vuông, với một số bóng để có thể đọc, bạn có thể làm:

path_effects = [patheffects.withSimplePatchShadow(shadow_rgbFace=(1,1,1))] 

for i, j in product(range(m), range(n)): 
    _ = ax.text(j, i, '{0:.2f}'.format(df.iloc[i, j]), 
       size='medium', ha='center', va='center', 
       path_effects=path_effects)