本文介绍基于Python语言,读取文件夹下大量栅格遥感影像文件,并基于给定的一个像元,提取该像元对应的全部遥感影像文件中,指定多个波段的数值;修改其中不在给定范围内的异常值,并计算像元数值在每一景遥感影像中变化的差值;最终将这些数据保存为一个新的Excel表格文件的方法。
首先,我们来看一下本文需要具体实现的需求。现在有一个文件夹,如下图所示;其中,存放了大量的遥感影像文件,且每一景遥感影像都是同一个空间位置、不同成像时间对应的遥感影像,因此其空间参考信息、栅格的行数与列数等都是一致的。此外,每一景遥感影像都具有5
个不同的波段。
我们现在希望,给定一个像元(也就是给定了这个像元在遥感影像中的行号与列号),提取出在指定的波段中(我们这里就提取全部的5
个波段),该像元对应的每一景遥感影像的数值(也就是提取了该像元在每一景遥感影像、每一个波段的数值);随后,将提取到的大于1
的数值修改为1
,并计算像素值在每一景遥感影像中数值的差值;最后,将提取到的数据保存为一个Excel表格文件。
明确了需求,我们就可以撰写代码;具体如下。
# -*- coding: utf-8 -*-
"""
Created on Thu Jul 27 11:25:55 2023
@author: fkxxgis
"""
import os
import pandas as pd
from osgeo import gdal
def extract_pixel_time_series(input_folder, output_csv):
tif_files = [file for file in os.listdir(input_folder) if file.endswith('.tif')]
target_row = 495
target_col = 60
time_series_df = pd.DataFrame()
for tif_file in tif_files:
file_path = os.path.join(input_folder, tif_file)
dataset = gdal.Open(file_path)
for band in range(dataset.RasterCount):
band_data = dataset.GetRasterBand(band + 1).ReadAsArray()
pixel_value = band_data[target_row, target_col]
date = tif_file[10: 24]
time_series_df.at[date, f'Band_{band + 1}'] = pixel_value
dataset = None
for index in range(len(time_series_df.columns)):
time_series_df = time_series_df.apply(lambda x: x.clip(upper = 1))
new_col_name = time_series_df.columns[index] + "_diff"
time_series_df[new_col_name] = time_series_df.iloc[:, index].diff()
time_series_df.to_csv(output_csv)
# 示例用法
input_folder = r"E:\01_Reflectivity\FiveBands"
output_csv = r"E:\01_Reflectivity\Data.csv"
extract_pixel_time_series(input_folder, output_csv)
首先,我们需要导入必要的模块和库。其中os
用于操作文件和文件夹,pandas
用于处理数据和创建DataFrame格式数据,而gdal
则用于读取栅格数据;关于gdal
库的配置方法,大家可以参考文章用whl文件配置Python库:Anaconda环境配置GDAL(https://www.jianshu.com/p/59eae2f2339a)。
随后,我们对extract_pixel_time_series
这个函数加以定义。这个函数接收两个参数input_folder
和output_csv
,分别表示存储栅格数据的文件夹路径和输出的Excel文件的路径。随后,列出input_folder
文件夹下所有以.tif
结尾的文件,并存储在列表中。其次,循环遍历每个栅格文件,构建完整的文件路径,用于后面的数据读取,并使用gdal.Open()
打开栅格文件,获取数据集对象。
接下来,通过循环遍历每个波段。读取当前波段的数据,并存储在band_data
变量中。随后基于我们给定的像元位置,提取目标像元的数值(位置就是这个[target_row, target_col]
)。此外,为了使得我们保存结果时可以记录每一个数值对应的成像日期,因此需要从文件名中提取日期,并存储在date
变量中。
接下来,通过time_series_df.at[date, f'Band_{band + 1}']
,将像元值存储在DataFrame中,行索引为日期,列名为Band_1
、Band_2
等;随后,将数据集对象dataset
设为None
,释放内存资源。
接下来,我们将大于1
的数值加以处理,并计算每个波段随时间变化的数值之差。遍历time_series_df
的每一列,并对于每一列使用clip(upper=1)
将超过1
的值截断为1
;随后,为每一列创建新列,列名为原列名加上_diff
,存储该列差值。
最后,我们将处理后的时间序列数据保存为Excel表格文件即可。
运行上述代码,我们即可获得多个遥感影像文件中,给定像元位置处,像元数值的时间变化序列,并可以获得其变化值。
至此,大功告成。