Plotar shapefile com matplotlib

Eu estou tentando ler um shapefile e plotá-lo usando matplotlib. Aqui está o código:

import matplotlib.pyplot as plt
import shapefile   

shpFilePath = "D:\test.shp"  
listx=[]
listy=[]
test = shapefile.Reader(shpFilePath)
for sr in test.shapeRecords():
    for xNew,yNew in sr.shape.points:
        listx.append(xNew)
        listy.append(yNew)
plt.plot(listx,listy)
plt.show()

No entanto, recebo linhas conectando meus polígonos. Como posso desenhar os polígonos de forma que eles sejam o caminho no arquivo de forma. Aqui estão as capturas de tela do gráfico e do shapefile quando ele é aberto com o ArcGIS.

Vou deixar para você como coletar as formas, mas este é o princípio

import numpy as np
from matplotlib import pyplot as p  #contains both numpy and pyplot
x1 = [-1,-1,10,10,-1]; y1 = [-1,10,10,-1,-1]
x2 = [21,21,29,29,21]; y2 = [21,29,29,21,21]
shapes = [[x1,y1],[x2,y2]]
for shape in shapes:
  x,y = shape
  p.plot(x,y)
p.show()

Para referências futuras, eis a solução que encontrei depois de seguir os conselhos acima.

import shapefile as shp  # Requires the pyshp package
import matplotlib.pyplot as plt

sf = shp.Reader("test.shp")

plt.figure()
for shape in sf.shapeRecords():
    x = [i[0] for i in shape.shape.points[:]]
    y = [i[1] for i in shape.shape.points[:]]
    plt.plot(x,y)
plt.show()

A figura resultante será muito colorida, mas você precisará ajustar as palavras-chave da plotagem.

Você precisa usar caminhos e patches matplotlib e existe um módulo Python dedicado a plotar polígonos de shapefiles usando essas funções Descartes .

Como o Pyshp (shapefile) possui a convenção geo_interface ( Nova geo_interface para PyShp ), você pode usá-lo.

polys  = shapefile.Reader("polygon")
# first polygon
poly = polys.iterShapes().next().__geo_interface__
print poly
{'type': 'Polygon', 'coordinates': (((151116.87238259654, 135890.8706318218), (153492.19971554304, 134793.3055883224), (153934.50204650551, 133892.31935858406), (152623.97662143156, 131811.86024627919), (150903.91200102202, 130894.49244872745), (149347.66305874675, 132991.33312884573), (149151.08424498566, 134383.76639298678), (151116.87238259654, 135890.8706318218)),)}

O resultado é a representação GeoJSON da geometria e você pode usar a solução de Como plotar dados geográficos usando matplotlib / python

import matplotlib.pyplot as plt 
from descartes import PolygonPatch
BLUE = '#6699cc'
fig = plt.figure() 
ax = fig.gca() 
ax.add_patch(PolygonPatch(poly, fc=BLUE, ec=BLUE, alpha=0.5, zorder=2 ))
ax.axis('scaled')
plt.show()

Isso pode ser feito usando geopandas ou pyshp, conforme discutido nesta resposta . Geopandas usam matplotlib em seu back-end para plotagem.

Além de responder a ldocao e responder à pergunta do FaCoffee. Quando você tiver ilhas isoladas e elas fizerem parte do mesmo recurso, poderá tentar a seguir:

import shapefile as shp
import matplotlib.pyplot as plt

sf = shp.Reader("test.shp")

plt.figure()
for shape in sf.shapeRecords():
    for i in range(len(shape.shape.parts)):
        i_start = shape.shape.parts[i]
        if i==len(shape.shape.parts)-1:
            i_end = len(shape.shape.points)
        else:
            i_end = shape.shape.parts[i+1]
        x = [i[0] for i in shape.shape.points[i_start:i_end]]
        y = [i[1] for i in shape.shape.points[i_start:i_end]]
        plt.plot(x,y)
plt.show()

Isso faz funcionar para mim. A propriedade "partes" de uma forma retorna os índices iniciais de diferentes geometrias dentro de um recurso.

Ainda assim, em um formato de arquivo de forma, pode haver várias partes. Isso plotará cada parte em uma forma, separadamente.

import matplotlib.pyplot as plt
import shapefile
import numpy as np

this_shapefile = shapefile.Reader(map_file_base) # whichever file
shape = this_shapefile.shape(i) # whichever shape
points = np.array(shape.points)

intervals = list(shape.parts) + [len(shape.points)]

ax = plt.gca()
ax.set_aspect(1)

for (i, j) in zip(intervals[:-1], intervals[1:]):
    ax.plot(*zip(*points[i:j]))