Spaces:
Sleeping
Sleeping
Carl Boettiger
commited on
Commit
•
705cf55
1
Parent(s):
f144e02
full raster-vector-extract
Browse files- raster-vector-extract.py +20 -1
raster-vector-extract.py
CHANGED
@@ -25,7 +25,7 @@ def extract(raster, vector, layer, output = None):
|
|
25 |
polys = ogr.Open(vrt)
|
26 |
exact_extract(rast, polys, 'mean', output='gdal', output_options = {'filename' : output })
|
27 |
return output
|
28 |
-
|
29 |
|
30 |
# -
|
31 |
|
@@ -48,6 +48,25 @@ rasters = [
|
|
48 |
]
|
49 |
# extract(rasters[0], vector, layer) # just one
|
50 |
|
|
|
|
|
|
|
|
|
51 |
parallel_extract = delayed(extract)
|
52 |
runner = Parallel(n_jobs=-1)
|
53 |
runner(parallel_extract(i, vector, layer) for i in rasters)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
polys = ogr.Open(vrt)
|
26 |
exact_extract(rast, polys, 'mean', output='gdal', output_options = {'filename' : output })
|
27 |
return output
|
28 |
+
|
29 |
|
30 |
# -
|
31 |
|
|
|
48 |
]
|
49 |
# extract(rasters[0], vector, layer) # just one
|
50 |
|
51 |
+
# +
|
52 |
+
# %%time
|
53 |
+
|
54 |
+
# extract all the raster layers in parallel
|
55 |
parallel_extract = delayed(extract)
|
56 |
runner = Parallel(n_jobs=-1)
|
57 |
runner(parallel_extract(i, vector, layer) for i in rasters)
|
58 |
+
|
59 |
+
# +
|
60 |
+
# bind columns into a single csv
|
61 |
+
csv_files = list(Path.cwd().glob("*.csv"))
|
62 |
+
|
63 |
+
import pandas as pd
|
64 |
+
data = {}
|
65 |
+
for file in csv_files:
|
66 |
+
df = pd.read_csv(file.name)
|
67 |
+
data[file.stem] = df['mean']
|
68 |
+
|
69 |
+
pd.DataFrame(data).to_csv("pad_raster_means.csv")
|
70 |
+
# -
|
71 |
+
|
72 |
+
[file.unlink() for file in csv_files]
|