pythonでシェープファイルを読み込んで出力する時に属性の定義を変えないようにしたい

Question

前提・実現したいこと

shpファイルをgeopandasで読み込んで出力する時に、属性定義を変えないようにしたい
ある程度は出来ていると思うのですが充分でないならもう少し改善したい。
また、他の方法等もあれば知りたいです。
よろしくお願いします。

該当のソースコード

import geopandas as gpd

data_frame=gpd.GeoDataFrame.from_file(r"C:\work\sample01\result\sample4.shp")

# これだけだと属性定義が変わる
data_frame.to_file(r"C:\work\sample01\result\sample4_2.shp")


# gdalで読み込んで定義を取得すると、属性定義が保持される
import osgeo.ogr as ogr
import osgeo.osr as osr

driver = ogr.GetDriverByName("ESRI Shapefile")
data_source = driver.Open(r"C:\work\sample01\result\sample4.shp")
layer=data_source.GetLayer()

## gdalとgeopandasの型の対応。これ対応あってます？足りない？。
## いや、これだと保持されないな…シェープファイルなら、これだけも要らないかな
dict_field_type={
ogr.OFTInteger:'int', # Simple 32bit integer.
ogr.OFTReal:'float', # Double Precision floating point.
ogr.OFTString:'str', # String of ASCII chars.
ogr.OFTWideString:'str', # deprecated
ogr.OFTDate:'date', # Date.
ogr.OFTTime:'time', # Time.
ogr.OFTDateTime:'datetime', # Date and Time.
ogr.OFTInteger64:'int', # Single 64bit integer.
}

## gdalで型定義を読み込み、geopandasのschemaの要素を作成
dict_type ={}
layer_defn = layer.GetLayerDefn()
for column in range(layer_defn.GetFieldCount()):
    field_defn=layer_defn.GetFieldDefn(column)
    str_name=field_defn.GetName()
    int_type=field_defn.GetType()
    int_width=field_defn.GetWidth()
    int_precision=field_defn.GetPrecision()
    if int_type == ogr.OFTReal:
        dict_type[str_name]=dict_field_type[int_type] + ':' + str(int_width) + '.' + str(int_precision)
    elif type in [ogr.OFTInteger,ogr.OFTInteger64,ogr.OFTString]:
        dict_type[str_name]=dict_field_type[int_type] + ':' + str(int_width)
    else:
        dict_type[str_name]=dict_field_type[int_type]

## schemaを作成
schema=gpd.io.file.infer_schema(data_frame)
for str_name in dict_type:
    schema['properties'][str_name]=dict_type[str_name]

## schemaを指定して出力
data_frame.to_file(r"C:\work\sample01\result\sample4_3.shp",schema=schema)

補足情報（FW/ツールのバージョンなど）

Python 3.9.2
GDAL 3.2.1
geopandas 0.8.2

OGRFieldTypeの一覧

ここにある全てを対応した方がいいと思うのですが
geopandas側での記述方法がわからないです…

Answer 1

オリジナルのシェープファイルからスキーマを取得して、geopandas からの書き出し時に渡す、という方法は同じなのですが、スキーマの取得を簡略化する、以下のような方法もあります。

import fiona
import geopandas as gpd

fp = r"C:\work\sample01\result\sample4.shp"
data_frame = gpd.read_file(fp)

with fiona.open(fp) as source:
    data_frame.to_file(r"C:\work\sample01\result\sample4_3.shp", schema=source.schema)

fiona は geopandas の依存ライブラリです。シェープファイルの読み込みは実際には fiona が行っています。fiona 自身も GDAL/OGR に依存しています。

fiona がシェープファイルを読み込んだ時点ではスキーマを保持しているのですが、geopandas オブジェクトが作られる時点で失われます。

ここでは改めて fiona でシェープファイルを開き直して、スキーマを取得しています。

Comments

[xail2222]

回答ありがとうございます。
fiona は、使ったことがなかったのですが説明は理解できました。
print(source.schema['properties'])
で想定通りの結果が返ってきました。
fionaも勉強していこうと思います。