只绘制相关矩阵的一行或几行

Question

我有一个名为corrdata的相关矩阵，我用numpy.corrcoef计算了这个矩阵。然后我要做的就是提取这个矩阵的一行或几行，现在只想绘制它们而不是整个矩阵。 因为矩阵不再是正方形的，所以不可能使用pcolor、imshow等来绘制数据。在

所以我想找一种最好的替代方法来绘制这些提取的相关系数，得到与相关矩阵相同的外观，用彩色正方形表示相关系数的值，但只显示几行完整的矩阵。在

Answer 1

您可以简单地插入一个额外的单例维度，以便将您的(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数组的更多方法。在