拼接右移算法

# -*- coding: cp936 -*-
import numpy as np
from osgeo import gdal
import numpy as np
path=r'J:\gz\gfvwf1\GF1_WFV2_E94.6_N41.0_20240811_L1A13520258001\maskp.dat'
# 打开ENVI文件
dataset = gdal.Open(path, gdal.GA_ReadOnly)

# 检查数据集是否成功打开
if dataset is None:
    raise FileNotFoundError(f"无法打开文件: {envi_file_path}")

# 获取第一个波段（通常ENVI文件是多波段的，但这里只读取一个作为示例）
band = dataset.GetRasterBand(1)

# 读取波段数据到NumPy数组
arr = band.ReadAsArray()
# 原始数组
# -*- coding: cp936 -*-
import numpy as np
# 获取数组的行数和列数
rows, cols = arr.shape

# 初始化一个空列表来存储满足条件的索引
indices = []
arr2 = np.zeros(arr.shape)
# 遍历数组中的每个元素（除了最后一行和最后一列）
for i in range(rows):
    for j in range(cols):
        # 检查当前元素是否为1
        if arr[i, j] == 1:
            # 检查右侧是否为1
            #arr[i, j]=0
            if j + 1 < cols and arr[i, j + 1] == 1  and i + 1 < rows and arr[i + 1, j] == 1:
                arr2[i+1, j + 1]=1
                
            # 检查下方是否为1
            elif j + 1 < cols and arr[i, j + 1] == 1  and i + 1 < rows and arr[i + 1, j] == 0:
                arr2[i + 1, j]=1
                
            elif j + 1 < cols and arr[i, j + 1] == 0  and i + 1 < rows and arr[i + 1, j] == 1:
                arr2[i, j + 1]=1
            elif j + 1 < cols and arr[i, j + 1] == 0  and i + 1 < rows and arr[i + 1, j] == 0:
                arr2[i, j + 1]=1
                arr2[i + 1, j]=1

# 打印满足条件的索引
print(arr2)