Carl Boettiger commited on
Commit
705cf55
1 Parent(s): f144e02

full raster-vector-extract

Browse files
Files changed (1) hide show
  1. 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]