rasterio批量裁剪栅格数据的指定区域：以裁剪掉tif的首行、保留其余行列为例

PI理论

于 2024-01-26 20:44:42 发布

阅读量497

点赞数 6

分类专栏： Python 文章标签： python

本文链接：https://blog.csdn.net/weixin_46080995/article/details/135874482

版权

Python 专栏收录该内容

57 篇文章 2 订阅

订阅专栏

本文介绍了如何使用Python库Rasterio对.tif文件进行批量裁剪，指定删除首行（y_start=1），保留其余行列，然后保存到自定义输出路径。方法涉及读取文件、创建窗口、裁剪数据和写入新文件。

摘要由CSDN通过智能技术生成

# -*- coding: utf-8 -*-
"""
rasterio批量裁剪栅格数据的指定区域，这里以裁剪掉tif的首行、保留其余行列为例
2024.01.12, 22:38, YMJ
"""

import os
import rasterio
from rasterio.windows import Window


def crop_tiff_with_metadata(input_path, output_path, x_start, y_start, x_end, y_end):
    with rasterio.open(input_path) as src:
        # 通过 window 参数指定裁剪区域
        window = Window(x_start, y_start, x_end - x_start, y_end - y_start)

        # 读取裁剪区域的数据
        data = src.read(window=window)

        # 获取裁剪区域的元数据
        meta = src.meta
        meta['width'], meta['height'] = window.width, window.height
        meta['transform'] = src.window_transform(window)

    # 将裁剪后的数据写入新的TIFF文件
    with rasterio.open(output_path, 'w', **meta) as dst:
        dst.write(data)

# 想要被批量裁剪的数据的存放路径
dir = "D:\\MODIS\\lc_resample"

# 读取路径下所有的栅格文件，并将其绝对路径存储到列表中
file_list = []
for root,subroot,files in os.walk(dir):
    for file in files:
        if file.endswith(".tif"):
            file_path = os.path.join(root, file)
            file_list.append(file_path)
# print(file_list)

# 遍历列表中的栅格文件
for i in range(len(file_list)):
    print(file_list[i])

    # 自定义输出路径
    outpath = "D:\\YRB\\SGCF\\" + "lc\\" + \
                  file_list[i].split("\\")[3]

    # 调用函数，617、386分别是列、行，(0 1 617 386)指的是裁剪掉第一行后剩下的区域范围
    crop_tiff_with_metadata(file_list[i], outpath, 0, 1, 617, 386)