Shapelyジオメトリをシェープファイルに書き込む方法


26

shapelyからshapefileにジオメトリデータ構造を書き込む簡単な方法を誰かが実証できますか 特に、穴と線ストリングのあるポリゴンに興味があります。また、arcpyから離れることも有益です(したがって、osgeo、pyshpなどがすべて優れています)。

回答:


44

よく知られているバイナリは、ShapelyやGDAL / OGRなどの多くのGISソフトウェアと交換できる優れたバイナリ交換形式です。

これは、ワークフローの小さな例ですosgeo.ogr

from osgeo import ogr
from shapely.geometry import Polygon

# Here's an example Shapely geometry
poly = Polygon([(0, 0), (0, 1), (1, 1), (0, 0)])

# Now convert it to a shapefile with OGR    
driver = ogr.GetDriverByName('Esri Shapefile')
ds = driver.CreateDataSource('my.shp')
layer = ds.CreateLayer('', None, ogr.wkbPolygon)
# Add one attribute
layer.CreateField(ogr.FieldDefn('id', ogr.OFTInteger))
defn = layer.GetLayerDefn()

## If there are multiple geometries, put the "for" loop here

# Create a new feature (attribute and geometry)
feat = ogr.Feature(defn)
feat.SetField('id', 123)

# Make a geometry, from Shapely object
geom = ogr.CreateGeometryFromWkb(poly.wkb)
feat.SetGeometry(geom)

layer.CreateFeature(feat)
feat = geom = None  # destroy these

# Save and close everything
ds = layer = feat = geom = None

更新:ポスターはGDAL / OGRの回答を受け入れましたが、フィオナに相当するものは次のとおりです。

from shapely.geometry import mapping, Polygon
import fiona

# Here's an example Shapely geometry
poly = Polygon([(0, 0), (0, 1), (1, 1), (0, 0)])

# Define a polygon feature geometry with one attribute
schema = {
    'geometry': 'Polygon',
    'properties': {'id': 'int'},
}

# Write a new Shapefile
with fiona.open('my_shp2.shp', 'w', 'ESRI Shapefile', schema) as c:
    ## If there are multiple geometries, put the "for" loop here
    c.write({
        'geometry': mapping(poly),
        'properties': {'id': 123},
    })

(Windowsユーザーに注意:言い訳はありません


Fionaライブラリではなくこの方法を選んだ理由に興味があります。
ネイサンW

1
さて、ポスターはosgeo.ogrの例を探していましたが、比較は興味深いものです。
13

@sgilliesの明示的な比較が追加されました
マイクT

3
まあ、正直に言うと、ほとんどが語用論でした。私は質問に答えてコードをデモンストレーションする努力に感謝し、すでにosgeoをいじっていました。それ以来、私は両方の方法を試しましたが、どちらも十分な答えです。レスポンダーの努力が正確かつ迅速であることに感謝します。
terra_matics

@Mike T osgeo.ogrアプローチに関しては、QGISのPythonプラグインで使用しています。考慮されるシェープファイルは、Line(ShapelyのLineString)です。「poly」変数を定義したところで、Qgs.Rectangleからの座標で「line」変数を定義しました。私は正確なコードを使用しましたが、エラーはありませんが、機能を追加せず、機能のないシェープファイルを提供します。
-Akhil

28

FionaはShapelyとうまく機能するように設計しました。これらを一緒に使用してシェープファイル機能を「クリーン」にする非常に簡単な例を次に示します。

import logging
import sys

from shapely.geometry import mapping, shape

import fiona

logging.basicConfig(stream=sys.stderr, level=logging.INFO)

with fiona.open('docs/data/test_uk.shp', 'r') as source:

    # **source.meta is a shortcut to get the crs, driver, and schema
    # keyword arguments from the source Collection.
    with fiona.open(
            'with-shapely.shp', 'w',
            **source.meta) as sink:

        for f in source:

            try:
                geom = shape(f['geometry'])
                if not geom.is_valid:
                    clean = geom.buffer(0.0)
                    assert clean.is_valid
                    assert clean.geom_type == 'Polygon'
                    geom = clean
                f['geometry'] = mapping(geom)
                sink.write(f)

            except Exception, e:
                # Writing uncleanable features to a different shapefile
                # is another option.
                logging.exception("Error cleaning feature %s:", f['id'])

https://github.com/Toblerity/Fiona/blob/master/examples/with-shapely.pyから。


6

PyShpを使用してShapelyジオメトリを作成することもできます(元のポスターでもPyShpについて質問されているため)。

1つの方法は、シェイプジオメトリをgeojsonに変換し(shapely.geometry.mappingメソッドを使用)、変更されたPyShpのフォークを使用し、シェープファイルに書き込むときにgeojsonジオメトリディクショナリを受け入れるWriterメソッドを提供することです。

メインのPyShpバージョンに依存する場合は、以下の変換関数も提供しています。

# THIS FUNCTION CONVERTS A GEOJSON GEOMETRY DICTIONARY TO A PYSHP SHAPE OBJECT
def shapely_to_pyshp(shapelygeom):
    # first convert shapely to geojson
    try:
        shapelytogeojson = shapely.geometry.mapping
    except:
        import shapely.geometry
        shapelytogeojson = shapely.geometry.mapping
    geoj = shapelytogeojson(shapelygeom)
    # create empty pyshp shape
    record = shapefile._Shape()
    # set shapetype
    if geoj["type"] == "Null":
        pyshptype = 0
    elif geoj["type"] == "Point":
        pyshptype = 1
    elif geoj["type"] == "LineString":
        pyshptype = 3
    elif geoj["type"] == "Polygon":
        pyshptype = 5
    elif geoj["type"] == "MultiPoint":
        pyshptype = 8
    elif geoj["type"] == "MultiLineString":
        pyshptype = 3
    elif geoj["type"] == "MultiPolygon":
        pyshptype = 5
    record.shapeType = pyshptype
    # set points and parts
    if geoj["type"] == "Point":
        record.points = geoj["coordinates"]
        record.parts = [0]
    elif geoj["type"] in ("MultiPoint","Linestring"):
        record.points = geoj["coordinates"]
        record.parts = [0]
    elif geoj["type"] in ("Polygon"):
        record.points = geoj["coordinates"][0]
        record.parts = [0]
    elif geoj["type"] in ("MultiPolygon","MultiLineString"):
        index = 0
        points = []
        parts = []
        for eachmulti in geoj["coordinates"]:
            points.extend(eachmulti[0])
            parts.append(index)
            index += len(eachmulti[0])
        record.points = points
        record.parts = parts
    return record

関数をコピーして独自のスクリプトに貼り付け、それを呼び出すだけで、任意の形状の形状をpyshp互換の形状に変換できます。それらを保存するには、結果の各pyshp形状をshapefile.Writerインスタンスの._shapesリストに追加するだけです(例については、この投稿の下部にあるテストスクリプトを参照してください)。

ただし、この関数は内部のポリゴンホールがあればそれを処理せず、単に無視します。その機能を機能に追加することは確かに可能ですが、私はまだ気にしていません。機能を改善するための提案や編集は大歓迎です:)

完全なスタンドアロンテストスクリプトを次に示します。

### HOW TO SAVE SHAPEFILE FROM SHAPELY GEOMETRY USING PYSHP

# IMPORT STUFF
import shapefile
import shapely, shapely.geometry

# CREATE YOUR SHAPELY TEST INPUT
TEST_SHAPELYSHAPE = shapely.geometry.Polygon([(133,822),(422,644),(223,445),(921,154)])

#########################################################
################## END OF USER INPUT ####################
#########################################################

# DEFINE/COPY-PASTE THE SHAPELY-PYSHP CONVERSION FUNCTION
def shapely_to_pyshp(shapelygeom):
    # first convert shapely to geojson
    try:
        shapelytogeojson = shapely.geometry.mapping
    except:
        import shapely.geometry
        shapelytogeojson = shapely.geometry.mapping
    geoj = shapelytogeojson(shapelygeom)
    # create empty pyshp shape
    record = shapefile._Shape()
    # set shapetype
    if geoj["type"] == "Null":
        pyshptype = 0
    elif geoj["type"] == "Point":
        pyshptype = 1
    elif geoj["type"] == "LineString":
        pyshptype = 3
    elif geoj["type"] == "Polygon":
        pyshptype = 5
    elif geoj["type"] == "MultiPoint":
        pyshptype = 8
    elif geoj["type"] == "MultiLineString":
        pyshptype = 3
    elif geoj["type"] == "MultiPolygon":
        pyshptype = 5
    record.shapeType = pyshptype
    # set points and parts
    if geoj["type"] == "Point":
        record.points = geoj["coordinates"]
        record.parts = [0]
    elif geoj["type"] in ("MultiPoint","Linestring"):
        record.points = geoj["coordinates"]
        record.parts = [0]
    elif geoj["type"] in ("Polygon"):
        record.points = geoj["coordinates"][0]
        record.parts = [0]
    elif geoj["type"] in ("MultiPolygon","MultiLineString"):
        index = 0
        points = []
        parts = []
        for eachmulti in geoj["coordinates"]:
            points.extend(eachmulti[0])
            parts.append(index)
            index += len(eachmulti[0])
        record.points = points
        record.parts = parts
    return record

# WRITE TO SHAPEFILE USING PYSHP
shapewriter = shapefile.Writer()
shapewriter.field("field1")
# step1: convert shapely to pyshp using the function above
converted_shape = shapely_to_pyshp(TEST_SHAPELYSHAPE)
# step2: tell the writer to add the converted shape
shapewriter._shapes.append(converted_shape)
# add a list of attributes to go along with the shape
shapewriter.record(["empty record"])
# save it
shapewriter.save("test_shapelytopyshp.shp")

5

カリムの答えはかなり古いですが、私は彼のコードを使用し、彼に感謝したかったです。彼のコードを使用して私が考え出した1つの小さなこと:形状タイプがPolygonまたはMultipolygonの場合、まだ複数のパーツ(内部の穴)がある可能性があります。したがって、彼のコードの一部を次のように変更する必要があります

elif geoj["type"] == "Polygon":
    index = 0
    points = []
    parts = []
    for eachmulti in geoj["coordinates"]:
        points.extend(eachmulti)
        parts.append(index)
        index += len(eachmulti)
    record.points = points
    record.parts = parts
elif geoj["type"] in ("MultiPolygon", "MultiLineString"):
    index = 0
    points = []
    parts = []
    for polygon in geoj["coordinates"]:
        for part in polygon:
            points.extend(part)
            parts.append(index)
            index += len(part)
    record.points = points
    record.parts = parts
弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.