我有一些dxf
文件,并希望将它们转换为geojson
文件:
import subprocess
from subprocess import call
import os
working_directory = 'D:/dxf_files/'
for subdir, dirs, files in os.walk(working_directory):
for file in files:
if file.endswith('.dxf'):
print(file)
输出:
BJ-SZZDS-1010084246-dongta-11.dxf
BJ-SZZDS-1010084246-dongta-12.dxf
BJ-SZZDS-1010084246-dongta-17.dxf
BJ-SZZDS-1010084246-dongta-18.dxf
BJ-SZZDS-1010084246-dongta-19.dxf
...
我想把这些文件放在下面的input_file
中,通过替换文件的扩展名保持output_file
文件名与input_file
相同。现在分开两个代码块,如何将它们组合在一起?感谢您提前的任何帮助。
input_file = 'BJ-SZZDS-1010084246-dongta-11.dxf'
output_file = 'BJ-SZZDS-1010084246-dongta-11.geojson'
def dxf2geojson(output_file, input_file):
command = ['ogr2ogr', '-f', 'GeoJSON', output_file, input_file]
p = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
return p
dxf2geojson(output_file, input_file)
首先,您可以将所有文件名保存在列表中,例如file_list
:
import subprocess
from subprocess import call
import os
working_directory = 'D:/dxf_files/'
file_list = [] # define file_list to save all dxf files
for subdir, dirs, files in os.walk(working_directory):
for file in files:
if file.endswith('.dxf'):
file_list.append(file) # save the filenames in file_list
然后,从file_list
执行每个文件:
def dxf2geojson(output_file, input_file):
command = ['ogr2ogr', '-f', 'GeoJSON', output_file, input_file]
p = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
return p
for input_file in file_list:
f = input_file[:-4] # to omit .dxf
output_file = f + '.geojson' # add file extension .geojson
dxf2geojson(output_file, input_file)
您可以通过将文件迭代代码中的打印功能替换为转换功能来完成此操作。
import subprocess
from subprocess import call
import os
working_directory = 'D:/dxf_files/'
for subdir, dirs, files in os.walk(working_directory):
for file in files:
if file.endswith('.dxf'):
input_file = file
output_file = file[:-3]+'geojson'
P = dxf2geojson(output_file, input_file)
您可以将所有文件保存到列表中,然后迭代它。
import subprocess
from subprocess import call
import os
working_directory = 'D:/dxf_files/'
def_list = []
for subdir, dirs, files in os.walk(working_directory):
for file in files:
if file.endswith('.dxf'):
dxf_list.append(file)
def dxf2geojson(output_file, input_file):
command = ['ogr2ogr', '-f', 'GeoJSON', output_file, input_file]
p = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
return p
for dex_file in dexf_list:
output_file = dex_file[:-4] + '.geojson'
dxf2geojson(output_file, dex_file)