腳本通過目錄中的文件循環

Question

我有以下代碼，它使用我需要的數據從shp.file創建需要的txt文件。我有一個名爲profile的文件夾，其中包含一些名爲（profil1.shp，profil2.shp，profil3.shp等）的形狀文件。我想知道如何創建一個循環，以便腳本爲每個文件創建一個具有相同名稱的txt文件（例如，用於profil1.shp創建profil1.txt，profil2.shp創建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()列出當前目錄中的文件和文件夾。 這將返回當前目錄中的所有文件的列表（或者給它的目錄作爲參數，如果沒有指定參數，它會檢查當前目錄）。

然後，您可以使用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，操作系統，SYS，OSR，os.path中

os.chdir（」 .. \配置文件） L = os.listdir（） 對於i在升： 如果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。