BUG: Allow using numpy in DataFrame.eval and DataFrame.query via @-notation

doc/source/whatsnew/v3.0.0.rst

@@ -325,6 +325,7 @@ Bug fixes
 - Fixed bug in :class:`SparseDtype` for equal comparison with na fill value. (:issue:`54770`)
 - Fixed bug in :meth:`.DataFrameGroupBy.median` where nat values gave an incorrect result. (:issue:`57926`)
 - Fixed bug in :meth:`DataFrame.cumsum` which was raising ``IndexError`` if dtype is ``timedelta64[ns]`` (:issue:`57956`)
+- Fixed bug in :meth:`DataFrame.eval` and :meth:`DataFrame.query` which caused an exception when using numpy via ``@`` notation. (:issue:`58041
 - Fixed bug in :meth:`DataFrame.join` inconsistently setting result index name (:issue:`55815`)
 - Fixed bug in :meth:`DataFrame.to_string` that raised ``StopIteration`` with nested DataFrames. (:issue:`16098`)
 - Fixed bug in :meth:`DataFrame.transform` that was returning the wrong order unless the index was monotonically increasing. (:issue:`57069`)

pandas/core/computation/ops.py

@@ -115,7 +115,7 @@ def _resolve_name(self):
         res = self.env.resolve(local_name, is_local=is_local)
         self.update(res)
 
-        if hasattr(res, "ndim") and res.ndim > 2:
+        if hasattr(res, "ndim") and isinstance(res.ndim, int) and res.ndim > 2:
             raise NotImplementedError(
                 "N-dimensional objects, where N > 2, are not supported with eval"
             )

pandas/tests/computation/test_eval.py

@@ -737,6 +737,13 @@ def test_and_logic_string_match(self):
         assert pd.eval(f"{event.str.match('hello').a}")
         assert pd.eval(f"{event.str.match('hello').a and event.str.match('hello').a}")
 
+    def test_using_numpy(self):
+        # GH 58041
+        df = Series([0.2, 1.5, 2.8], name="a").to_frame()
+        res = df.eval("@np.floor(a)")
+        expected = np.floor(df["a"])
+        tm.assert_series_equal(expected, res)
+
 
 # -------------------------------------
 # gh-12388: Typecasting rules consistency with python