僅繪製一個或幾行相關矩陣

Question

我有一個相關矩陣，名爲corrdata，我使用numpy.corrcoef進行計算。然後我要做的是提取這個矩陣的一行或幾行，現在只想繪製它們而不是整個矩陣。 因爲矩陣不再是正方形，所以不可能使用pcolor,imshow等來繪製數據。

所以我想要求最好的替代方法來繪製這些提取的相關係數，並得到與相關矩陣相同的外觀，用相關係數值的彩色方塊表示，但只顯示幾行完整的矩陣。

Answer 1

你可以簡單地插入一個額外的單維度，以將您的(n,) 1D載體引入(1, n)二維數組，然後用pcolor，imshow等正常：

import numpy as np 
from matplotlib import pyplot as plt 

# dummy correlation coefficients 
coeffs = np.random.randn(10, 10) 
row = coeffs[0] 

# indexing with None (or equivalently, np.newaxis) inserts an extra singleton 
# dimension 
plt.imshow(row[None, :], cmap=plt.cm.jet, interpolation='nearest') 

See here爲將一維矢量轉換爲二維數組的更多方法。