Вы можете использовать такой же режим расширения с параметром оси, как 1
>>> import numpy as np
>>> import matplotlib.pyplot as plt
>>> a = np.array([1, 2, 3, 4])
>>> a.shape
(4,)
>>> plt.imshow(np.expand_dims(a, axis=1)); plt.show()
работает как положено.