仅绘制一个或几行相关矩阵

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为将一维矢量转换为二维数组的更多方法。