栅格数据创建与保存

2024-05-02 14:18
文章标签 保存 创建 栅格数据

本文主要是介绍栅格数据创建与保存,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

栅格数据创建与保存

作者:阿振

邮箱:tanzhenyugis@163.com

博客:https://blog.csdn.net/theonegis/article/details/80089375

修改时间:2018-05-24

声明:本文为博主原创文章,转载请注明原文出处


思路与方法

使用Python进行栅格数据处理,很多时候,我们会将GDAL的Dataset对象转化为NumPy的ndarray对象,这样我们可以使用很多通用的Python库对数据进行处理,然后再借助GDAL库将数据写回到文件。

不同于普通的二进制文件,空间栅格数据的写需要注意两点:

  1. 数据的投影信息(确定了平面坐标系)
  2. 数据的地理坐标信息(确定了图像在给定坐标系下的位置)

在GDAL中,我们首先需要创建Dataset对象,然后给Dataset对象填充数据以及元数据信息就OK了。

Driver或者说GDALDriver(Python版本的API中对象名称好像都去掉了前缀GDAL,而C/C++版本的API很多对象前面都是有GDAL前缀的,如GDALDataset对象在Python中对应的是Dataset对象)有两个方法:Create()CreateCopy()

所以,相应地,我们也有两种思路去创建一个Dataset对象:

  1. 如果我们有一个原型数据,比如我们对原始数据进行了处理,处理之后,空间信息,波段等都没有变化,则可以将原始数据作为原型数据,使用CreateCopy()方法创建一个和原始数据一样的Dataset对象,然后在创建好的对象中填充一个ndarray数据就好了。
  2. 如果我们没有一个原型数据,那么我们首先需要使用Create()方法创建一个空的Dataset对象,然后手动设置对象的波段,尺寸,空间信息等,然后再在对应的波段填空ndarray具体的数据。

实现函数

我把上面两种实现思路编码成一个函数,具体实现如下:

def array2raster(f_name, np_array, driver='GTiff',prototype=None,xsize=None, ysize=None,transform=None, projection=None,dtype=None, nodata=None):"""将ndarray数组写入到文件中:param f_name: 文件路径:param np_array: ndarray数组:param driver: 文件格式驱动:param prototype: 文件原型:param xsize: 图像的列数:param ysize: 图像的行数:param transform: GDAL中的空间转换六参数:param projection: 数据的投影信息:param dtype: 数据存储的类型:param nodata: NoData元数据"""# 创建要写入的数据集(这里假设只有一个波段)# 分两种情况:一种给定了数据原型,一种没有给定,需要手动指定Transform和Projectiondriver = gdal.GetDriverByName(driver)if prototype:dataset = driver.CreateCopy(f_name, prototype)else:if dtype is None:dtype = gdal.GDT_Float32if xsize is None:xsize = np_array.shape[-1]  # 数组的列数if ysize is None:ysize = np_array.shape[-2]  # 数组的行数dataset = driver.Create(f_name, xsize, ysize, 1, dtype)  # 这里的1指的是一个波段dataset.SetGeoTransform(transform)dataset.SetProjection(projection)# 将array写入文件dataset.GetRasterBand(1).WriteArray(np_array)if nodata is not None:dataset.GetRasterBand(1).SetNoDataValue(nodata)dataset.FlushCache()return f_name

在使用该函数的时候,要么传进去一个prototype原型数据集,要么传进去transformprojection等信息,这样写入的文件才具有空间参考。

测试案例

下面是一个计算NDVI(Normalized Difference Vegetation Index,归一化植被指数)和DVI(Difference Vegetation Index,差值植被指数)的例子。我们首先计算NDVI,然后通过从原始数据中读取的空间投影和空间变换六元组信息创建输出文件;然后再计算DVI,通过NDVI文件作为原型数据集,以创建DVI的输出数据集。

具体实现如下:

# 打开栅格数据集
ds = gdal.Open('example.tif') # example.tif有三个波段,分别是蓝,红,近红外# 获取数据集的一些信息
x_size = ds.RasterXSize  # 图像列数
y_size = ds.RasterYSize  # 图像行数proj = ds.GetProjection()  # 返回的是WKT格式的字符串
trans = ds.GetGeoTransform()  # 返回的是六个参数的tuple# 在数据集层面ReadAsArray方法将每个波段都转换为了一个二维数组
image = ds.ReadAsArray()# 获得波段对应的array
bnd_red = image[1].astype(float)  # 红波段
bnd_nir = image[2].astype(float)  # 近红外波段idx_ndvi = (bnd_nir - bnd_red) / (bnd_nir + bnd_red)  # 计算NDVI指数out1_file = 'NDVI.tif'
array2raster(out1_file, idx_ndvi,xsize=x_size, ysize=y_size,transform=trans, projection=proj,dtype=gdal.GDT_Float32)idx_dvi = bnd_nir - bnd_red  # 计算DVI指数out2_file = 'DVI.tif'
# 这里我们使用out1_file作为原型图像作为参考来保存out2_file
array2raster(out2_file, idx_ndvi, prototype=gdal.Open(out1_file))# 关闭数据集
ds = None

这篇关于栅格数据创建与保存的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/954443

相关文章

创建Java keystore文件的完整指南及详细步骤

《创建Javakeystore文件的完整指南及详细步骤》本文详解Java中keystore的创建与配置,涵盖私钥管理、自签名与CA证书生成、SSL/TLS应用,强调安全存储及验证机制,确保通信加密和... 目录1. 秘密键(私钥)的理解与管理私钥的定义与重要性私钥的管理策略私钥的生成与存储2. 证书的创建与

python如何创建等差数列

《python如何创建等差数列》:本文主要介绍python如何创建等差数列的问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录python创建等差数列例题运行代码回车输出结果总结python创建等差数列import numpy as np x=int(in

怎么用idea创建一个SpringBoot项目

《怎么用idea创建一个SpringBoot项目》本文介绍了在IDEA中创建SpringBoot项目的步骤,包括环境准备(JDK1.8+、Maven3.2.5+)、使用SpringInitializr... 目录如何在idea中创建一个SpringBoot项目环境准备1.1打开IDEA,点击New新建一个项

如何使用Maven创建web目录结构

《如何使用Maven创建web目录结构》:本文主要介绍如何使用Maven创建web目录结构的问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录创建web工程第一步第二步第三步第四步第五步第六步第七步总结创建web工程第一步js通过Maven骨架创pytho

MySQL 用户创建与授权最佳实践

《MySQL用户创建与授权最佳实践》在MySQL中,用户管理和权限控制是数据库安全的重要组成部分,下面详细介绍如何在MySQL中创建用户并授予适当的权限,感兴趣的朋友跟随小编一起看看吧... 目录mysql 用户创建与授权详解一、MySQL用户管理基础1. 用户账户组成2. 查看现有用户二、创建用户1. 基

Java使用HttpClient实现图片下载与本地保存功能

《Java使用HttpClient实现图片下载与本地保存功能》在当今数字化时代,网络资源的获取与处理已成为软件开发中的常见需求,其中,图片作为网络上最常见的资源之一,其下载与保存功能在许多应用场景中都... 目录引言一、Apache HttpClient简介二、技术栈与环境准备三、实现图片下载与保存功能1.

Python中使用uv创建环境及原理举例详解

《Python中使用uv创建环境及原理举例详解》uv是Astral团队开发的高性能Python工具,整合包管理、虚拟环境、Python版本控制等功能,:本文主要介绍Python中使用uv创建环境及... 目录一、uv工具简介核心特点:二、安装uv1. 通过pip安装2. 通过脚本安装验证安装:配置镜像源(可

Java中实现线程的创建和启动的方法

《Java中实现线程的创建和启动的方法》在Java中,实现线程的创建和启动是两个不同但紧密相关的概念,理解为什么要启动线程(调用start()方法)而非直接调用run()方法,是掌握多线程编程的关键,... 目录1. 线程的生命周期2. start() vs run() 的本质区别3. 为什么必须通过 st

Macos创建python虚拟环境的详细步骤教学

《Macos创建python虚拟环境的详细步骤教学》在macOS上创建Python虚拟环境主要通过Python内置的venv模块实现,也可使用第三方工具如virtualenv,下面小编来和大家简单聊聊... 目录一、使用 python 内置 venv 模块(推荐)二、使用 virtualenv(兼容旧版 P

Linux lvm实例之如何创建一个专用于MySQL数据存储的LVM卷组

《Linuxlvm实例之如何创建一个专用于MySQL数据存储的LVM卷组》:本文主要介绍使用Linux创建一个专用于MySQL数据存储的LVM卷组的实例,具有很好的参考价值,希望对大家有所帮助,... 目录在Centos 7上创建卷China编程组并配置mysql数据目录1. 检查现有磁盘2. 创建物理卷3. 创