python - 从 landsat tar.gz 列表中提取 2 个特定影像波段

Question

我有一个 Landsat 场景列表。tar.gz 内部有多个波段。喜欢：

文件“ LC08018051date.tar.gz ”

LC08_date_sr_band1.tif
LC08_date_sr_band2.tif
LC08_date_sr_band3.tif
LC08_date_sr_band4.tif
LC08_date_sr_band5.tif
LC08_date_sr_band6.tif
LC08_date_sr_band7.tif

我只需要提取LC08_date_sr_band3.tif并LC08_date_sr_band4.tif计算一个索引。

此代码可以从所有文件中提取所有“乐队”。

import tarfile
import os
import glob
files = glob.glob('L*.tar.gz')
for fileName in files:
    tfile = tarfile.open(fileName, 'r:gz')
    membersList = tfile.getmembers()
    namesList = tfile.getnames()
    bandsList = [x for x, y in zip(membersList, namesList) if "sr_band" in y]
    print("extracting...")
    tfile.extractall("Folder/",members=bandsList)
    print ("Done")

问题是我只需要提取 sr_band3 和 sr_band4

Answer 1

不是最好的选择，但我所做的是：

import tarfile
import os
import glob
import re
files = glob.glob('L*.tar.gz')
if not files: 
   raise Exception('No se han encontrado archivos a procesar')
for fileName in files:
    tfile = tarfile.open(fileName, 'r:gz')
    membersList = tfile.getmembers()
    namesList = tfile.getnames()
    bandsList = [x for x, y in zip(membersList, namesList) if "sr_band3" in y]
    print("extracting...")
    tfile.extractall("bands/",members=bandsList)
    print ("Done")
for fileName in files:
    tfile = tarfile.open(fileName, 'r:gz')
    membersList = tfile.getmembers()
    namesList = tfile.getnames()
    bandsList = [x for x, y in zip(membersList, namesList) if "sr_band4" in y]
    print("extracting...")
    tfile.extractall("bands/",members=bandsList)
    print ("Done")

代码每次都输入到文件中，所以不是最快的方法。