"""
    在执行提取操作前要先对与之无关的影像区域进行遮罩。这种技术被称为阈值分割（二值化）。本例中，
阈值图像上岛屿和背景区域的水域形成了足够鲜明的对比，阈值化（二值化）后能够完美地将它分隔开来。
"""
from osgeo import gdal_array


src = "TIF/islands/islands.tif"
output = "TIF/islands/islands_classified.tiff"
srcArray = gdal_array.LoadFile(src)

# 将直方图分割成2个子区间以便分类
classes = gdal_array.numpy.histogram(srcArray, bins=2)[1]
lut = [[250, 0, 0], [0, 0, 0], [255, 255, 255]]

# 分类起始值
start = 1

# 建立输出图片
rgb = gdal_array.numpy.zeros((3, srcArray.shape[0], srcArray.shape[1],), gdal_array.numpy.float32)

# 处理所有类别并分配颜色
for i in range(len(classes)):
    mask = gdal_array.numpy.logical_and(start <= srcArray, srcArray <= classes[i])
    for j in range(len(lut[i])):
        rgb[j] = gdal_array.numpy.choose(mask, (rgb[j], lut[i][j]))
    start = classes[i] + 1

# 保存图片
gdal_array.SaveArray(rgb.astype(gdal_array.numpy.uint8), output, format="GTIFF", prototype=src)