Rasterio快速入门指南：Python栅格数据处理基础

Rasterio快速入门指南：Python栅格数据处理基础

【免费下载链接】rasterio Rasterio reads and writes geospatial raster datasets 项目地址: https://gitcode.com/gh_mirrors/ra/rasterio

概述

Rasterio是一个强大的Python库，专门用于处理栅格地理空间数据。它基于GDAL构建，提供了更Pythonic的接口来处理各种栅格数据格式。本文将带你快速了解如何使用Rasterio进行基本的栅格数据读写操作。

环境准备

在开始之前，请确保你已经安装了Rasterio库。本文所有示例都假设你已经在Python环境中正确安装了Rasterio。

读取栅格数据

打开数据集

读取栅格数据的第一步是打开数据集。Rasterio使用open()函数来打开栅格文件：

import rasterio
dataset = rasterio.open('example.tif')

这里的example.tif是一个示例GeoTIFF文件，包含Landsat 8的波段4数据。

数据集属性

打开数据集后，我们可以访问其各种属性：

print(dataset.name)    # 文件路径
print(dataset.mode)    # 打开模式('r'表示只读)
print(dataset.closed)  # 是否已关闭
print(dataset.count)   # 波段数量
print(dataset.width)   # 图像宽度(列数)
print(dataset.height)  # 图像高度(行数)
print(dataset.dtypes)  # 各波段的数据类型

地理参考信息

栅格数据通常包含地理参考信息，这些信息定义了像素如何映射到地球表面：

# 获取数据边界框
print(dataset.bounds)

# 获取坐标参考系统(CRS)
print(dataset.crs)  # 输出可能类似: EPSG:32612

# 获取仿射变换矩阵
transform = dataset.transform
print(transform)

仿射变换矩阵非常重要，它将像素坐标(col, row)转换为地理坐标(x, y)。例如：

# 左上角像素的地理坐标
print(transform * (0, 0))

# 右下角像素的地理坐标
print(transform * (dataset.width, dataset.height))

读取栅格数据

读取波段数据

要读取特定波段的数据，可以使用read()方法：

# 读取第一个波段(波段索引从1开始)
band1 = dataset.read(1)

这将返回一个NumPy数组，包含该波段的所有像素值。

空间索引

Rasterio提供了方便的方法在像素坐标和地理坐标之间转换：

# 地理坐标转像素坐标
x, y = (dataset.bounds.left + 100000, dataset.bounds.top - 50000)
row, col = dataset.index(x, y)

# 像素坐标转地理坐标
geo_x, geo_y = dataset.xy(row, col)

创建和写入栅格数据

准备数据

首先，我们需要创建要写入的数据。以下示例创建了一个包含两个高斯分布差异的数组：

import numpy as np

x = np.linspace(-4.0, 4.0, 240)
y = np.linspace(-3.0, 3.0, 180)[::-1]
X, Y = np.meshgrid(x, y)
Z1 = np.exp(-2 * np.log(2) * ((X - 0.5)**2 + (Y - 0.5)**2) / 1**2)
Z2 = np.exp(-3 * np.log(2) * ((X + 0.5)**2 + (Y + 0.5)**2) / 2.5**2)
Z = 10.0 * (Z2 - Z1)

创建新数据集

要创建新的栅格文件，我们需要以写入模式('w')打开数据集，并提供必要的参数：

from rasterio.transform import Affine

# 计算分辨率和变换矩阵
res = (x[-1] - x[0]) / 240.0
transform = Affine.translation(x[0] - res/2, y[0] + res/2) * Affine.scale(res, -res)

# 打开新数据集
with rasterio.open(
    'output.tif',
    'w',
    driver='GTiff',
    height=Z.shape[0],
    width=Z.shape[1],
    count=1,
    dtype=Z.dtype,
    crs='+proj=latlong',  # 经纬度坐标系统
    transform=transform,
) as dst:
    dst.write(Z, 1)  # 将数据写入第一个波段

关键参数说明

• driver: 指定输出格式(如'GTiff'表示GeoTIFF)
• height和width: 栅格的行数和列数
• count: 波段数量
• dtype: 数据类型(如'float64')
• crs: 坐标参考系统
• transform: 仿射变换矩阵

最佳实践

1. 使用上下文管理器: 如示例所示，使用with语句可以确保文件正确关闭。

2. 数据类型选择: 根据数据特性选择合适的数据类型，以节省存储空间。

3. 坐标系统: 确保为数据设置正确的坐标参考系统。

4. NoData值: 对于包含无效值的数据，设置合适的nodata值。

总结

本文介绍了Rasterio的基本使用方法，包括：

• 如何打开和读取现有栅格文件
• 如何访问栅格数据的属性和地理参考信息
• 如何在像素坐标和地理坐标之间转换
• 如何创建新的栅格文件并写入数据

掌握了这些基础知识后，你就可以开始使用Rasterio进行更复杂的栅格数据处理任务了。对于更高级的功能，如波段运算、重投影、窗口读取等，可以参考Rasterio的进阶文档。

【免费下载链接】rasterio Rasterio reads and writes geospatial raster datasets 项目地址: https://gitcode.com/gh_mirrors/ra/rasterio