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