从tiff图像中查找NDVI索引时,在python脚本中出现错误

时间:2015-11-05 14:10:29

标签: python indexoutofboundsexception

我正在努力将NDVI索引从图像转换为数字,以便我可以打印NDVI数值。

为此目的,我在google搜索并发现了一个python代码。我正在使用Ubuntu,我安装了pyhton 2和3。

我对python很新。

# Import the Python 4 print function
from __future__ import print_function

# Import the "gdal" and "gdal_array" submodules from within the "osgeo" module
from osgeo import gdal
from osgeo import gdal_array

# Import the NumPy module
import numpy as np

# Open a GDAL dataset
dataset = gdal.Open('/home/path to directory/A0_200_T.TIF', gdal.GA_ReadOnly)

# Allocate our array using the first band's datatype
image_datatype = dataset.GetRasterBand(1).DataType

image = np.zeros((dataset.RasterYSize, dataset.RasterXSize, dataset.RasterCount),
                 dtype=gdal_array.GDALTypeCodeToNumericTypeCode(image_datatype))

# Loop over all bands in dataset
for b in xrange(dataset.RasterCount):
    # Remember, GDAL index is on 1, but Python is on 0 -- so we add 1 for our GDAL calls
    band = dataset.GetRasterBand(b + 1)

    # Read in the band's data into the third dimension of our array
    image[:, :, b] = band.ReadAsArray()


print('Red band mean: {r}'.format(r=image[:, :, 2].mean()))
print('NIR band mean: {nir}'.format(nir=image[:, :, 3].mean()))
b_red = 2
b_nir = 3

ndvi = (image[:, :, b_nir] - image[:, :, b_red]) / \
        (image[:, :, b_nir] + image[:, :, b_red]).astype(np.float64)

print('NDVI matrix: ')
print(ndvi)

print('\nMax NDVI: {m}'.format(m=ndvi.max()))
print('Mean NDVI: {m}'.format(m=ndvi.mean()))
print('Median NDVI: {m}'.format(m=np.median(ndvi)))
print('Min NDVI: {m}'.format(m=ndvi.min()))

现在当我使用" python name.py"运行此脚本时它导致我出现以下错误。

Traceback (most recent call last):
  File "ndvi.py", line 29, in <module>
    print('Red band mean: {r}'.format(r=image[:, :, 5].mean()))
IndexError: index 5 is out of bounds for axis 2 with size 1

似乎数组存储了超出绑定的数据,但我不知道如何解决这个问题。(!??)

我也不知道它给出了NDVI指数的天气,但我希望如此。(请建议我是否有另一种方法可以这样做。)

对此的任何帮助都会非常明显。

如果您需要更多信息,请与我们联系。

提前谢谢你。 R上。

0 个答案:

没有答案