Python:将多个Landsat图像导入Grass Gis的脚本

Question

我正在尝试通过调整此代码来编写一个将Landsat卫星图像导入Grass GIS的Python代码：>

Landsat瓷砖作为文件夹下载，每个文件夹包含7个TIFF图像（Band 1-7）。因此，我有一个包含几个子目录的目录（每个Landsat瓷砖一个）。

目前我的代码如下：

#!/usr/bin/python
import os
import sys
import glob
import grass.script as grass
def import_tifs(dirpath):
    for dirpath, dirname, filenames in os.walk(dirpath):
        for dirname in dirpath:
        dirname = os.path.join(dirpath,dirname)
            for file in os.listdir(dirname):
                if os.path.splitext(file)[-1] != '.TIF':
                    continue
                ffile = os.path.join(dirname, file)
                name = os.path.splitext(file)[0].split(dirname)[-1]
                grass.message('Importing %s -> %s@%s...' % (file, name, dirpath))
                grass.run_command('r.in.gdal',
                                  flags = 'o',
                                  input = ffile,
                                  output = name,
                                  quiet = True,
                                  overwrite = True)
def main():                                 
    if len(sys.argv) == 1:
        for directory in filter(os.path.isdir, os.listdir(os.getcwd())):
            import_tifs(directory)
    else:
        import_tifs(sys.argv[1])
if __name__ == "__main__":
    main()

我会收到以下错误：

Traceback (most recent call last):
  File "C:/Users/Simon/Documents/import_landsat2.py", line
40, in <module>
    main()
  File "C:/Users/Simon/Documents/import_landsat2.py", line
37, in main
    import_tifs(sys.argv[1])
  File "C:/Users/Simon/Documents/import_landsat2.py", line
17, in import_tifs
    for file in os.listdir(dirname):
WindowsError: [Error 3] The system cannot find the path
specified: 'dirpath\C/*.*'

任何人都可以解释正在发生的事情以及我需要做什么来修复它，还是建议替代方案？谢谢。

Answer 1

我相信您的主要问题是os.walk()中的dirname返回列表（而不是字符串），因此您后续的字符串（即dirname = os.path.join(dirpath,dirname)）有些畸形。这是一种可能的替代方法 - 为了进行测试，我将目录的完整路径用作sys.argv[1]，但是您可以使其更具动态性以适合您的情况。另外，避免使用诸如file之类的变量名称，因为它们是Python关键字。我无法测试您的grass.*功能，但是希望这将是一个清晰的例子，以便您可以调整所需的方式。os.walk()本地处理许多标准解析，因此您可以删除一些目录 - 操纵功能：

def import_tifs(dirpath):
  for dirpath, dirname, filenames in os.walk(dirpath):
    # Iterate through the files in the current dir returned by walk()
    for tif_file in filenames:
      # If the suffix is '.TIF', process
      if tif_file.upper().endswith('.tif'):
        # This will contain the full path to your file
        full_path = os.path.join(dirpath, tif_file)
        # tif_file will already contain the name, so you can call from here
        grass.message('Importing %s -> %s@%s...' % (full_path, tif_file, dirpath))
        grass.run_command('r.in.gdal',
                          flags = 'o',
                          input = full_path,
                          output = tif_file,
                          quiet = True,
                          overwrite = True)

Answer 2

我刚刚重写了您的代码以列出所有dir树并找到文件扩展名，在这种情况下，

#!/usr/bin/python
import os
import sys

def import_tifs(dirpath):
    for dirpath, dirname, filenames in os.walk(dirpath):
        for filename in filenames:
            name, extension = os.path.splitext(filename)
            if extension.lower() == ".tif":
                filepath = os.path.join(dirpath, filename)
                print(filepath, name, dirpath)

def main():
    if len(sys.argv) == 1:
            import_tifs(os.getcwd())
    else:
        import_tifs(sys.argv[1])

if __name__ == "__main__":
    main()

请检查您是否正在寻找...