将 dem 作为数组加载

Question

我写了一个 cod，它应该将 dem 读取为 np 数组，然后我想绘制数组。它没有绘制我想要的图像，而是绘制了纯黄色图像。这是代码：

import os
from osgeo import gdal
import matplotlib.pyplot as plt
ds = gdal.Open('dem.tif')
gt = ds.GetGeoTransform()
prj = ds.GetProjection()
band = ds.GetRasterBand(1)
array = band.ReadAsArray()
plt.figure()
plt.imshow(array)

我找遍了都找不到问题所在