Demo

Author: tomwhite

demo.py

@@ -0,0 +1,33 @@
+import xarray as xr
+import numpy.array_api as xp
+if __name__ == "__main__":
+
+    print("Create an xarray DataArray with a NumPy Array API array")
+    arr = xr.DataArray(xp.ones((2, 3)), dims=("x", "y"), coords={"x": [10, 20]})
+    print(arr)
+    print()
+
+    print("The underlying data is a NumPy Array API array")
+    print(type(arr.data))
+    print(arr.data)
+    print()
+
+    print("Perform an xarray arithmetic op (add)")
+    s = arr + 7
+    print(s)
+    print()
+
+    print("Perform an xarray reduction operation (sum)")
+    s = arr.sum(skipna=False)
+    print(s)
+    print()
+
+    print("Perform an xarray indexing operation")
+    s = arr[:, 0]
+    print(s)
+    print()
+
+    print("Transpose")
+    s = arr.transpose()
+    print(s)
+    print()