1. 全部读取tif文件代码
import os
import csv
import pandas as pd
from osgeo import gdal
size=0.5
file_chdir=os.path.join(os.getcwd(),'data','data_tif')
for root,dirs,files in os.walk(file_chdir):
for file in files:
suffix_name=os.path.splitext(file)[1]
if suffix_name=='.tif':
name,year=file.split('.')[0].split('_')
file_name='data/data_tif/'+file
open_name='data/data_tif_csv/'+file.split('.')[0]+'.csv'
print(file_name)
f=open(open_name, 'w', newline='',encoding='utf-8')
writer = csv.writer(f)
writer.writerow(['date','lon', 'lat','grid_row', 'grid_col',name])
dataset = gdal.Open(file_name)
band=dataset.RasterCount
col = dataset.RasterXSize
row = dataset.RasterYSize
geo_information = dataset.GetGeoTransform()
for day in range(1,band+1):
dem = dataset.GetRasterBand(day).ReadAsArray()
for y in range(row):
rows = []
for x in range(col):
if dem[y][x] > 0:
lon = geo_information[0] + x * geo_information[1] + y * geo_information[2]
lat = geo_information[3] + x * geo_information[4] + y * geo_information[5]
writer.writerow([date,lon, lat, y, x, dem[y][x]])
2. 分块读取tif文件代码
2.1 位置读取tif文件
value=dataset.GetRasterBand(1).ReadAsArray(x,y,1,1)[0][0]
2.2 完整代码
import os
import csv
import pandas as pd
from osgeo import gdal
size=0.5
file_chdir=os.path.join(os.getcwd(),'data','data_tif')
for root,dirs,files in os.walk(file_chdir):
for file in files:
suffix_name=os.path.splitext(file)[1]
if suffix_name=='.tif':
name,year=file.split('.')[0].split('_')
file_name='data/data_tif/'+file
open_name='data/data_tif_csv/'+file.split('.')[0]+'.csv'
print(file_name)
f=open(open_name, 'w', newline='',encoding='utf-8')
writer = csv.writer(f)
writer.writerow(['date','lon', 'lat','grid_row', 'grid_col',name])
dataset = gdal.Open(file_name)
band=dataset.RasterCount
col = dataset.RasterXSize
row = dataset.RasterYSize
geo_information = dataset.GetGeoTransform()
for day in range(1,band+1):
dem = dataset.GetRasterBand(day).ReadAsArray(x,y,1,1)[0][0]
if dem>=0:
lon = geo_information[0] + x * geo_information[1] + y * geo_information[2]
lat = geo_information[3] + x * geo_information[4] + y * geo_information[5]
writer.writerow([date,lon, lat, y, x, dem[y][x]])
f.close()