Commit 22ffb5fb authored by Jordi Inglada's avatar Jordi Inglada
Browse files

ENH: return DEM as a 3 channel numpy array

parent 78e1588d
......@@ -60,7 +60,11 @@ class DEM:
elevation: np.ndarray
slope: np.ndarray
aspect: np.ndarray
transform: rio.transform.Affine
crs: Optional[str]
transform: Optional[rio.transform.Affine]
def as_stack(self):
return np.stack([self.elevation, self.slope, self.aspect], axis=0)
@dataclass(frozen=True)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment