使用matplotlib.pyplot.imshow() 顯示圖像時出現「TypeError: Invalid dimensions for image data」的問題

轉載自：https://blog.csdn.net/qq_17478877/article/details/88977295，本文只做個人記錄學習使用，版權歸原作者所有。 如上兩幅圖，matplotlib.pyplot.imshow()需要數據是二維的數組或者第三維深度是3或4的三維數組，當第三維深度爲1時，使用np.squeeze()壓縮數據成爲二維數組

>>阅读原文<<