gdal.打开索引器错误:列表索引超出范围

2024-06-11 05:26:09 发布

您现在位置:Python中文网/ 问答频道 /正文

我是新来的,希望能找到一些帮助。在

我使用下面的代码从Sentinel 2-A计算NDVI

import glob
import numpy as np
from osgeo import gdal 

in_dir = 'C:\\temp\\'

# Search directory for desired bands
red_file = glob.glob(in_dir + '**B04.jp2') # red band
nir_file = glob.glob(in_dir + '**B08.jp2') # nir band

# Define a function to calculate NDVI using band arrays for red, NIR bands
def ndvi(red, nir):
    return ((nir - red)/(nir + red))
    #return (NIR.astype(float) - RED.astype(float)) / (NIR+RED)


# Open each band using gdal
red_link = gdal.Open(red_file[0])
nir_link = gdal.Open(nir_file[0])

# read in each band as array and convert to float for calculations
red = red_link.ReadAsArray().astype(np.float)
nir = nir_link.ReadAsArray().astype(np.float)

# Call the ndvi() function on red, NIR bands
ndvi2 = ndvi(red, nir)

# Create output filename based on input name
outfile_name = red_file[0].split('_B')[0] + '_NDVI.tif'

x_pixels = ndvi2.shape[0] # number of pixels in x
y_pixels = ndvi2.shape[1] # number of pixels in y

# Set up output GeoTIFF
driver = gdal.GetDriverByName('GTiff')

# Create driver using output filename, x and y pixels, # of bands, and datatype
ndvi_data = driver.Create(outfile_name,x_pixels, y_pixels, 1,gdal.GDT_Float32)

# Set NDVI array as the 1 output raster band
ndvi_data.GetRasterBand(1).WriteArray(ndvi2)

# Setting up the coordinate reference system of the output GeoTIFF
geotrans=red_link.GetGeoTransform() # Grab input GeoTranform information
proj=red_link.GetProjection() # Grab projection information from input file

# now set GeoTransform parameters and projection on the output file
ndvi_data.SetGeoTransform(geotrans)
ndvi_data.SetProjection(proj)
ndvi_data.FlushCache()
ndvi_data=None
###############################################################################

我得到以下索引器:

^{pr2}$

有人知道如何修正这个错误吗?可能是因为我的GDAL版本不能识别jp2格式吗?在

谢谢!在


Tags: theinoutputdatabandlinkredfloat
1条回答
网友
1楼 · 发布于 2024-06-11 05:26:09

你说得对,glob的输出是空的!如果我将目录定义为:

# Search directory for desired bands
red_file = glob.glob(in_dir + '**B04.jp2') # red band
nir_file = glob.glob(in_dir + '**B08.jp2') # nir band

为了测试它,我用以下方法解决它:

^{pr2}$

但现在我得到了一个问题,即gdal.Open并没有按预期工作。在gdal.Open之后,下面的错误消息中又出现了一个空列表:

Traceback (most recent call last):
  File "C:\temp\NDVI3.py", line 52, in <module>
    red = red_link.ReadAsArray().astype(np.float)
AttributeError: 'NoneType' object has no attribute 'ReadAsArray'

现在我又问自己,是不是我的GDAL版本(2.2.3)不能识别jp2格式?在

相关问题 更多 >