Spaces:
Build error
Build error
Modularize python code
Browse files
plot.py
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import matplotlib.pyplot as plt
|
| 2 |
+
import geopandas as gpd
|
| 3 |
+
|
| 4 |
+
# all plotting functions
|
| 5 |
+
def create_plot(filename, extent, *gdfs): # takes in unlimited number of gdfs
|
| 6 |
+
fig, ax = plt.subplots(figsize=(10, 8)) #Sets image size by width & height (in inches)
|
| 7 |
+
|
| 8 |
+
colors = ['tan', 'mediumseagreen', 'thistle', 'lightcoral', 'sienna', 'yellow'] # Extend/improve this list as needed
|
| 9 |
+
|
| 10 |
+
for idx, gdf in enumerate(gdfs):
|
| 11 |
+
gdf.plot(ax=ax, color=colors[idx % len(colors)]) # Cycle through colors
|
| 12 |
+
|
| 13 |
+
ax.set_xlim(extent[0], extent[2])
|
| 14 |
+
ax.set_ylim(extent[1], extent[3])
|
| 15 |
+
|
| 16 |
+
# Hide axes
|
| 17 |
+
ax.axis('off')
|
| 18 |
+
|
| 19 |
+
plt.savefig(filename, bbox_inches='tight', pad_inches=0) # remove padding
|