脚本循环遍历目录中的文件

Question

我有以下代码，它使用我需要的数据从 shp.file 创建我需要的 txt 文件。我有一个名为配置文件的文件夹，其中包含一些名为（profil1.shp、profil2.shp、profil3.shp 等）的形状文件。我想知道如何创建一个循环，以便脚本为每个文件创建一个同名的 txt 文件（例如，for profil1.shp create profil1.txt、profil2.shp create profil2.txt等等）。

import ogr, os, sys, osr

os.chdir('..\profiles')

file = open('profil1.txt', 'w')

driver = ogr.GetDriverByName('ESRI Shapefile')

datasource = driver.Open('profil1.shp', 0)
if datasource is None:
  print 'Could not open file'
  sys.exit(1)

layer = datasource.GetLayer()

feature = layer.GetNextFeature()
while feature:

  id = feature.GetFieldAsString('ID')
  Distanta = feature.GetFieldAsString('DIST')
  Z = feature.GetFieldAsString('Z')
  geom = feature.GetGeometryRef()
  x = str(geom.GetX())
  y = str(geom.GetY())

  file.write(id + " " + Distanta + " " + "[X]:" + " " + x + ' ' + '[Y]:' + " "  + y + " " + " " + "[Z]" + Z + " " +  "\n")

  feature.Destroy()
  feature = layer.GetNextFeature()

datasource.Destroy()
file.close()

编辑：代码返回无法打开文件。Photo of the folder containing the files and their respective names. Safe to assume I am doing something wrong.

import ogr, os, sys, osr,os.path
os.chdir = ('C:\Users\Andrei\Desktop\profil3')
l = os.listdir('C:\Users\Andrei\Desktop\profil3')
for i in l:
    if i.endswith('.shp'):
        s1 = s.split('.')[0] + '.txt'
        file = open(s1, 'w')

    driver = ogr.GetDriverByName('ESRI Shapefile')
    datasource = driver.Open(i, 0)
    if datasource is None:
        print 'Could not open file'
        sys.exit(1)

    layer = datasource.GetLayer()

    feature = layer.GetNextFeature()
    while feature:

        id = feature.GetFieldAsString('ID')
        Distanta = feature.GetFieldAsString('DIST')
        Z = feature.GetFieldAsString('Z')
        geom = feature.GetGeometryRef()
        x = str(geom.GetX())
        y = str(geom.GetY())

        file.write(id + " " + Distanta + " " + "[X]:" + " " + x + ' ' + '[Y]:' + " "  + y + " " + " " + "[Z]" + Z + " " +  "\n")

        feature.Destroy()
        feature = layer.GetNextFeature()

    datasource.Destroy()
    file.close()

Answer 1

您可以使用os.listdir()列出当前目录中的文件和文件夹。 returns 当前目录（或作为参数给定的目录，如果未指定参数，则检查当前目录）中所有文件的列表。

然后您可以使用 string.endswith() 函数检查名称以 .shp 结尾的文件，然后使用它来创建新文件。

小部分示例-

import os , os.path
l = os.listdir()
for i in l:
    if i.endswith('.shp'):
        s1 = s.split('.')[0] + '.txt'

最后 s1 将包含扩展名为 .txt 的文件。

然后你可以在这个文件上做你的逻辑，并继续这样做。

完整代码类似于 -

导入 ogr，os，sys，osr，os.path

os.chdir('..\profiles') l = os.listdir() 对于我在我：如果 i.endswith('.shp'): s1 = s.split('.')[0] + '.txt' 文件=打开（s1，'w'）

    driver = ogr.GetDriverByName('ESRI Shapefile')

    datasource = driver.Open(i, 0)
    if datasource is None:
        print 'Could not open file'
        sys.exit(1)

    layer = datasource.GetLayer()

    feature = layer.GetNextFeature()
    while feature:

        id = feature.GetFieldAsString('ID')
        Distanta = feature.GetFieldAsString('DIST')
        Z = feature.GetFieldAsString('Z')
        geom = feature.GetGeometryRef()
        x = str(geom.GetX())
        y = str(geom.GetY())

        file.write(id + " " + Distanta + " " + "[X]:" + " " + x + ' ' + '[Y]:' + " "  + y + " " + " " + "[Z]" + Z + " " +  "\n")

        feature.Destroy()
        feature = layer.GetNextFeature()

    datasource.Destroy()
    file.close()

打开文件等的更好方法是使用 with 语句。查看它的教程 here。

脚本循环遍历目录中的文件

Script Loop through files in directory

python

ogr