fix: adding coordinate features

x,y,z coordinate features are useful for plotting.
Loop3D · Oct 9, 2024 · 464934b · 464934b
1 parent 9a32b4c
commit 464934b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/LoopStructural/utils/features.py b/LoopStructural/utils/features.py
@@ -0,0 +1,5 @@
+from ..modelling.features import LambdaGeologicalFeature
+
+X = LambdaGeologicalFeature(lambda pos: pos[:, 0], name="x")
+Y = LambdaGeologicalFeature(lambda pos: pos[:, 1], name="y")
+Z = LambdaGeologicalFeature(lambda pos: pos[:, 2], name="z")