Python：繪制大型矩陣的熱圖

Question

我有一個尺寸為500 X 18904的大型矩陣。

由於大多數值都是零，所以我無法清楚地看到圖案，因為零在顏色欄中占主導地位。

為了更仔細地查看數據，我需要放大圖像的不同部分。 有沒有可靠的方法可以使用色條顯示此數據？

這是我的代碼和輸出。

import numpy as np
import matplotlib.pyplot as plt
import scipy.io as sio
j = sio.loadmat('UV_matrix.mat')
k = j['UV']  
plt.imshow(k, aspect='auto')
plt.show()

輸出

Answer 1

我可以通過使用numpy數組想到兩個選擇。

1. 假設您的數據大部分都大於零，但是有很多零。

    vmin = some_value_higher_than_zero plt.matshow(k,aspect='auto',vmin=vmin) 

2. 將所有零設置為NaN。 他們被自動排除在外。

    k[k==0.0]=np.nan plt.matshow(k,aspect='auto') 

NB。 imshow和matshow都可以在這里使用。

當矩陣真正稀疏時，另一種選擇是使用散點圖。

x,y = k.nonzero()
plt.scatter(x,y,s=100,c=k[x,y]) #color as the values in k matrix