 |
Adding a field where there wasn't one before has
limitless possibilities. |
Today's example shows you how to use pyshp to add a new field to an existing shapefile. This operation is a two-step process. You must first update the dbf header to define the new field. Then you must update each record to account for a new column in the database so everything is balanced.
In the past, I've demonstrated modifying existing shapefiles for other reasons including merging shapefiles and deleting features in shapefiles. In every case you are actually reading in the existing shapefile, creating a new shapefile in memory and then writing out the new file either separately or on top of the old one. Even in really high-end GIS packages that's basically all you're doing. Some packages will use a temporary file in between.
Here's the example. We'll create a counter that gives us unique sample data to append to each record just so we can see the changes clearly. In the real world, you'd probably just insert a blank palce holder.
import shapefile
# Read in our existing shapefile
r = shapefile.Reader("Mississippi")
# Create a new shapefile in memory
w = shapefile.Writer()
# Copy over the existing fields
w.fields = list(r.fields)
# Add our new field using the pyshp API
w.field("KINSELLA", "C", "40")
# We'll create a counter in this example
# to give us sample data to add to the records
# so we know the field is working correctly.
i=1
# Loop through each record, add a column. We'll
# insert our sample data but you could also just
# insert a blank string or NULL DATA number
# as a place holder
for rec in r.records():
rec.append(i)
i+=1
# Add the modified record to the new shapefile
w.records.append(rec)
# Copy over the geometry without any changes
w._shapes.extend(r.shapes())
# Save as a new shapefile (or write over the old one)
w.save("Miss")
So there you have it. Overall it's a pretty simple process that can be extended to do some sophisticated operations. The sample Mississippi shapefile can be found here. But this shapefile only has one record so it's not that interesting. But it's lightweight and easy to examine the dbf file in your favorite spreadsheet program.
