我有一个名为corrdata
的相关矩阵,我使用numpy.corrcoef
计算。然后我做的是提取这个矩阵的一行或几行,现在只想绘制它们而不是整个矩阵。
由于矩阵不再是正方形,因此无法使用pcolor
,imshow
或类似图表绘制数据。
所以我想要求最好的替代方法来绘制这些提取的相关系数,并获得与相关矩阵在表示相关系数的值的彩色方块方面相同的外观,但仅显示几行完整矩阵。
答案 0 :(得分:0)
您可以简单地插入额外的单一维度,以便将(n,)
1D向量转换为(1, n)
2D数组,然后正常使用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了解将1D向量转换为2D数组的更多方法。