python - python matplotlib imshow在数据数组中具有不同的长度

Question

有没有办法使用 matplotlib 绘制具有不同行长的热图？

像这样：

plt.imshow( [ [1,2,3], [1,2], [1,2,3,4,5,6,7], [1,2,3,4]])
plt.jet()
plt.colorbar()
plt.show()

所需图像

Answer 1

鉴于所需的图像，我认为你会想要你plt.pcolormesh而不是imshow但我可能是错的。无论如何，我个人会创建一个函数来填充数组，然后使用掩码来绘制imshow或pcolormesh不绘制这些点。例如

import matplotlib.pylab as plt
import numpy as np

def regularise_array(arr, val=-1):
    """ Takes irregular array and returns regularised masked array

    This first pads the irregular awway *arr* with values *val* to make 
    it of rectangular. It then applies a mask so that the padded values
    are not displayed by pcolormesh. For this reason val should not
    be in *arr* as you will loose these points.
    """

    lengths = [len(d) for d in data]
    max_length = max(lengths)
    reg_array = np.zeros(shape=(arr.size, max_length))

    for i in np.arange(arr.size):
        reg_array[i] = np.append(arr[i], np.zeros(max_length-lengths[i])+val)

    reg_array = np.ma.masked_array(reg_array, reg_array == val)

    return reg_array

data = np.array([[1,2,3], [1,2], [1,2,3,4,5,6,7], [1,2,3,4]])

reg_data = regularise_array(data, val=-1)

plt.pcolormesh(reg_data)
plt.jet()
plt.colorbar()
plt.show()

这个问题是你需要注意val它不在数组中。您可以为此添加一个简单的检查或基于您正在使用的数据。for 循环可能会被矢量化，但我无法弄清楚如何。