ArcPyを使用して、同じ文字で始まるファイルをマージします
Aug 20 2020
以下と同じ2文字で始まるファイルをマージしようとしています。
- 11111-マージ(出力名:11.shp)
- 11112-マージ(出力名:11.shp)
- 11113-マージ(出力名:11.shp)
- 22112-マージ(出力名:22.shp)
- 22153-マージ(出力名:22.shp)
しかし、「11」がファイル名に含まれている場合(4など)、Pythonスクリプトの結果はマージされます。11 ***。shp(1)〜3))だけが必要ですが、スクリプトは11 .shpをマージします(1〜4を11.shpとしてマージします)
================================================== ======================
import sys, os, arcpy, fnmatch
reload(sys)
arcpy.env.workspace = r"F:\a"
workspace = arcpy.env.workspace
outdir = r"F:\b"
List = []
for dirpath, dirnames, filenames in os.walk(workspace, topdown=True):
for filename in filenames:
List.append(os.path.join(dirpath, filename))
break
patterns = set([os.path.basename(fl)[:2] for fl in List])
for pattern in patterns:
group = [fl for fl in List if fnmatch.fnmatchcase(fl, '*' + pattern + '*.shp')]
output = os.path.join(outdir, pattern)
arcpy.Merge_management(group, output)
================================================== ======================
どうすれば修正できますか?
回答
2 BERA Aug 20 2020 at 14:58
私はcollections.defaultdict(list)が好きです:
...キーと値のペアのシーケンスをリストの辞書にグループ化する
例:
import arcpy, os
from collections import defaultdict as dd
folder = r'C:\GIS\data\testdata\merge'
out_folder = r'C:\GIS\data\testdata\outfolder'
d = dd(list)
for item in os.listdir(folder):
if os.path.isfile(os.path.join(folder,item)) and file.endswith('.shp'):
d[item[:2]].append(os.path.join(folder, item))
#Execute code above then print(d) to make sure its ok before merging
for group, mergelist in d.items():
#print(group, mergelist)
arcpy.Merge_management(inputs=mergelist, output=os.path.join(out_folder, 'Merge_{}.shp'.format(group)))
サブディレクトリも検索する場合は、試したようにos.walkを使用します。例:
import os
from collections import defaultdict as dd
folder_main = r'C:\GIS\data\testdata'
d = dd(list)
for root, folder, files in os.walk(folder_main):
for file in files:
if os.path.isfile(os.path.join(root, file)) and file.endswith('.shp'):
print(file)
d[file[:2]].append(os.path.join(root, file))