2016-01-22 59 views
0

我有一個類似於此格式的數據集X = [[1,4,5], [34,70,1,5], [43,89,4,11], [22,76,4]]其中元素列表的長度不相等。創建一個具有不平衡行和列的棋盤圖

我想創建一個4行4列的棋盤圖,每個單位框的顏色條對應於數字的值。在這個數據集中,一些小盒子將會丟失(例如,第四列第一排)。

我該如何在python中使用matplotlib繪製這個圖? 謝謝

回答

2

您可以使用seaborn庫或matplotlib生成熱圖。首先,將其轉換爲熊貓數據框以處理缺失值。

import pandas as pd 
df = pd.DataFrame([[1,4,5],[34,70,1,5], [43,89,4,11],[22,76,4]]) 
%matplotlib inline 
from matplotlib import pyplot as plt 
import seaborn as sns 
sns.heatmap(df) 
plt.show() 

結果看起來像這樣。 enter image description here