一、两个文件。实现目标:从1个shp文件,根据属性表内容提取成200个shp文件,文件名取自txt文本内容
shp文件(要素1-200.shp):打开属性表前14项相同,后200项不相同。
————任务目标:需要输出200个shp(属性表前14项相同,循环后200项单独提取出1项)
txt文本(numbered.txt)内容:
1 要素1
2 要素2
... ...(总共200行)
import geopandas as gpd
# 文件路径
file_path = r"D:\要素\要素1-200.shp"
species_names_file = r"D:\numbered.txt"
# 读取要素名称
species_dict = {}
with open(species_names_file, 'r', encoding='utf-8') as file:
for line in file:
if line.strip():
number, name = line.strip().split(maxsplit=1)
species_dict[int(number)] = f"{number} {name}" # 存储编号和名称,包括空格
# 读取shapefile
data = gpd.read_file(file_path)
# 确定几何列的名称
geometry_column = data.geometry.name
# 获取前14个基础要素的列名
base_columns = data.columns[:14].tolist()
if geometry_column not in base_columns:
base_columns.append(geometry_column) # 确保几何列被包括
# 循环处理每个要素
for i in range(1, 200): # 物种编号从1到200
species_column = f'要素{i}' # 构建要素列名
if species_column in data.columns:
# 选择基础要素和当前要素的列,确保包括几何列
selected_columns = base_columns + [species_column]
selected_data = data[selected_columns]
# 使用包括编号和名称的字符串作为文件名
if i in species_dict:
filename = species_dict[i] # 从字典中获取包含编号和名称的字符串
new_file_path = fr"D:\要素\{filename}.shp"
else:
filename = f"{i} 要素_selected" # 如果编号不存在于字典中,使用默认名称
new_file_path = fr"D:\要素\{filename}.shp"
# 导出到新的shapefile,使用UTF-8编码
selected_data.to_file(new_file_path, encoding='utf-8')
print(f"{filename}的数据已成功导出到 {new_file_path}")
二、最后每个要素需要放在单独的文件夹中,文件夹名命名为文件名
文件名 文件夹名
1 要素1.shp 1 要素1
2 要素2.shp 2 要素2
......
import os
import shutil
# 包含 shapefile 文件的目录
source_dir = r"D:\要素\要素1-200"
# 遍历源目录中的所有文件
for filename in os.listdir(source_dir):
if filename.endswith(".shp"): # 检查文件是否为 shapefile
# 去掉文件扩展名得到基本名称
base_name = os.path.splitext(filename)[0]
# 使用基本名称创建新的目录路径
new_dir = os.path.join(source_dir, base_name)
# 如果目录不存在则创建目录
if not os.path.exists(new_dir):
os.makedirs(new_dir)
# 移动 shapefile 及其关联文件
for ext in ['.shp', '.shx', '.dbf', '.prj', '.cpg', '.sbn', '.sbx', '.qix', '.fbn', '.fbx', '.ain', '.aih', '.ixs', '.mxs', '.atx']:
old_file = os.path.join(source_dir, f"{base_name}{ext}")
if os.path.exists(old_file):
shutil.move(old_file, new_dir)
print(f"已将 {filename} 及其关联文件移动到 {new_dir}")