Added named_arrays.AbstractArray.cell_centers(), a method to conver…

…t from cell vertices to cell centers. (#93)

named_arrays/_core.py

@@ -455,6 +455,36 @@ def combine_axes(
         Array with the specified axes combined
         """
 
+    def cell_centers(
+        self,
+        axis: None | str | Sequence[str] = None,
+    ) -> na.AbstractExplicitArray:
+        """
+        Convert an array from cell vertices to cell centers.
+
+        Parameters
+        ----------
+        axis
+            The axes of the array to average over.
+        """
+
+        if axis is None:
+            axis = self.axes
+        elif isinstance(axis, str):
+            axis = (axis, )
+
+        result = self.explicit
+
+        shape = result.shape
+
+        for a in axis:
+            if a in shape:
+                lower = {a: slice(None, ~0)}
+                upper = {a: slice(+1, None)}
+                result = (result[lower] + result[upper]) / 2
+
+        return result
+
     def volume_cell(self, axis: None | str | Sequence[str]) -> na.AbstractScalar:
         """
         Computes the n-dimensional volume of each cell formed by interpreting

named_arrays/_functions/functions.py

@@ -128,6 +128,16 @@ def combine_axes(
             outputs=outputs.combine_axes(axes=axes, axis_new=axis_new),
         )
 
+    def cell_centers(
+        self,
+        axis: None | str | Sequence[str] = None,
+    ) -> na.AbstractExplicitArray:
+        return dataclasses.replace(
+            self,
+            inputs=self.inputs.cell_centers(axis),
+            outputs=self.outputs.cell_centers(axis),
+        )
+
     def to_string_array(
         self,
         format_value: str = "%.2f",

named_arrays/tests/test_core.py

@@ -257,6 +257,33 @@ def test_combine_axes(
             with pytest.raises(ValueError):
                 array.combine_axes(axes=axes, axis_new=axis_new)
 
+    @pytest.mark.parametrize(
+        argnames="axis",
+        argvalues=[
+            None,
+            "y",
+            ("y",),
+            ("x", "y"),
+        ]
+    )
+    def test_cell_centers(
+        self,
+        array: na.AbstractArray,
+        axis: None | str | Sequence[str],
+    ):
+        if axis is None:
+            axis_normalized = array.axes
+        elif isinstance(axis, str):
+            axis_normalized = (axis, )
+        else:
+            axis_normalized = axis
+
+        result = array.cell_centers(axis)
+
+        for a in axis_normalized:
+            if a in array.shape:
+                assert result.shape[a] == array.shape[a] - 1
+
     @pytest.mark.parametrize(
         argnames="axis",
         argvalues=[