利用python批量将.shp文件转换坐标生成.geojson文件,再将.geojson转换成.csv文件,最后将csv文件插入数据库表

第一步:.shp批量转.geojson

# author: JMY
# 创建时间: 2024/2/26 17:12
# 批量将.shp文件生成geojson文件并转换坐标为3857

import os
import geopandas as gpd

# 定义输入和输出文件夹路径
input_folder = 'shp文件'
output_folder = 'geojson文件'

# 定义输入和输出坐标系
out_proj = 'EPSG:3857'

# 获取输入文件夹下所有的 Shapefile 文件
shapefiles = [f for f in os.listdir(input_folder) if f.endswith('.shp')]

# 循环处理每个 Shapefile 文件
for index, shapefile in enumerate(shapefiles):
    print('转换第%d条数据...' % (index+1))
    # 读取 Shapefile 文件
    gdf = gpd.read_file(os.path.join(input_folder, shapefile))

    # 对几何数据进行坐标系转换
    gdf.to_crs(out_proj, inplace=True)

    # 构造输出文件名,将 '.shp' 替换为 '.geojson'
    output_file = os.path.splitext(shapefile)[0] + '.geojson'

    # 保存为 GeoJSON 文件
    gdf.to_file(os.path.join(output_folder, output_file), driver='GeoJSON')

print('Shapefile 转换为包含 EPSG:3857 坐标系的 GeoJSON 完成!')

第二步:.geojson批量转.csv

# author: JMY
# 创建时间: 2024/2/23 15:16
# 批量将geojson文件生成csv文件

import os
import geopandas as gpd
import json

# 定义一个函数,用于从几何信息中提取坐标
def extract_coordinates(geom):
    if geom:
        return json.dumps(geom.__geo_interface__['coordinates'])
    else:
        return None

# 输入目录和输出目录路径
input_dir = 'geojson文件'  # GeoJSON 文件所在目录路径
output_dir = 'csv文件'  # CSV 文件保存目录路径

# 获取输入目录下的所有 GeoJSON 文件名
file_names = [f for f in os.listdir(input_dir) if f.endswith('.geojson')]

# 初始化ID计数器
id_counter = 0

for index,file_name in enumerate(file_names):
    print('插入第%d条数据...' % (index+1))
    input_file = os.path.join(input_dir, file_name)

    # 读取 GeoJSON 数据并创建 GeoDataFrame 对象
    data = gpd.read_file(input_file)

    # 在DataFrame中插入一个自增的'id'列作为第一列,并累加ID计数器
    data.insert(0, 'id', range(1 + id_counter, len(data) + id_counter + 1))
    data.insert(1, 'cun_id', -1)

    # 提取 'coordinates' 和 'type' 信息
    data['type'] = data['geometry'].apply(lambda geom: geom.geom_type if geom else None)
    data['coordinates'] = data['geometry'].apply(extract_coordinates)

    # 删除原始的 'geometry' 列
    data = data.drop('geometry', axis=1)

    # 更新ID计数器
    id_counter += len(data)

    # 将所有字段中的空值设置为 null
    data = data.fillna(value='null')

    # 指定输出文件名(与原始文件同名,只改变后缀名)
    output_file = os.path.splitext(os.path.basename(input_file))[0] + '.csv'
    output_full_path = os.path.join(output_dir, output_file)

    # 导出为 CSV 文件
    data.to_csv(output_full_path, index=False)

print("输出完成......")

第三步:.csv批量插入数据库表

# author: JMY
# 创建时间: 2024/2/23 16:35
# 多个csv文件导入数据库同一张表中

import os
import glob
import pandas as pd
import mysql.connector

# 建立与MySQL数据库的连接
conn = mysql.connector.connect(
    host='', # 主机ip
    user='', # 账号
    password='', # 密码
    database='' # 数据库
)

# CSV目录路径
csv_directory = 'csv文件'

# 获取目录中的所有CSV文件
csv_files = glob.glob(os.path.join(csv_directory, '*.csv'))

# 创建游标对象
cursor = conn.cursor()

# 循环处理每个CSV文件
for index,csv_file in enumerate(csv_files):
    print('插入第%d条数据...'%(index+1))
    # 读取CSV文件
    data = pd.read_csv(csv_file, encoding='utf-8')  # 根据实际情况指定编码

    # 将除了'id'和'cun_id'外的其他字段转换为字符串类型
    for column in data.columns:
        if column not in ['id','cun_id']:
            data[column] = data[column].astype(str)

    # 构建SQL插入语句
    table_name = 'village_yjjbnt'  # 表名
    columns = ', '.join(data.columns)
    values = ', '.join(['%s'] * len(data.columns))
    insert_query = f"INSERT INTO {table_name} ({columns}) VALUES ({values})"

    # 批量插入数据
    records = data.values.tolist()
    cursor.executemany(insert_query, records)

# 提交事务
conn.commit()

# 关闭游标和连接
cursor.close()
conn.close()


print("数据导入mysql成功...")

  • 7
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

一只大菜鸟J

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值