Enabled PLW2901 on initial files under pandas/core/*

Author: ErdiTk

pandas/core/apply.py

@@ -1379,10 +1379,10 @@ def relabel_result(
         # mean  1.5
         # mean  1.5
         if reorder_mask:
-            fun = [
+            fun_name = [
                 com.get_callable_name(f) if not isinstance(f, str) else f for f in fun
             ]
-            col_idx_order = Index(s.index).get_indexer(fun)
+            col_idx_order = Index(s.index).get_indexer(fun_name)
             s = s[col_idx_order]
 
         # assign the new user-provided "named aggregation" as index names, and reindex
@@ -1421,14 +1421,16 @@ def _managle_lambda_list(aggfuncs: Sequence[Any]) -> Sequence[Any]:
     if len(aggfuncs) <= 1:
         # don't mangle for .agg([lambda x: .])
         return aggfuncs
-    i = 0
+
     mangled_aggfuncs = []
-    for aggfunc in aggfuncs:
-        if com.get_callable_name(aggfunc) == "<lambda>":
-            aggfunc = partial(aggfunc)
-            aggfunc.__name__ = f"<lambda_{i}>"
-            i += 1
-        mangled_aggfuncs.append(aggfunc)
+    for idx, aggfunc in enumerate(aggfuncs):
+        if not com.get_callable_name(aggfunc) == "<lambda>":
+            mangled_aggfuncs.append(aggfunc)
+            continue
+
+        partial_aggfunc = partial(aggfunc)
+        partial_aggfunc.__name__ = f"<lambda_{idx}>"
+        mangled_aggfuncs.append(partial_aggfunc)
 
     return mangled_aggfuncs
 

pandas/core/arrays/period.py

@@ -1099,7 +1099,7 @@ def _range_from_fields(
         freqstr = freq.freqstr
         year, quarter = _make_field_arrays(year, quarter)
         for y, q in zip(year, quarter):
-            y, m = parsing.quarter_to_myear(y, q, freqstr)
+            _, m = parsing.quarter_to_myear(y, q, freqstr)
             val = libperiod.period_ordinal(y, m, 1, 1, 1, 1, 0, 0, base)
             ordinals.append(val)
     else:

pandas/core/computation/eval.py

@@ -320,8 +320,8 @@ def eval(
     first_expr = True
     target_modified = False
 
-    for expr in exprs:
-        expr = _convert_expression(expr)
+    for expr_obj in exprs:
+        expr = _convert_expression(expr_obj)
         _check_for_locals(expr, level, parser)
 
         # get our (possibly passed-in) scope

pandas/core/computation/pytables.py

@@ -568,8 +568,7 @@ def __init__(
                 if isinstance(w, PyTablesExpr):
                     local_dict = w.env.scope
                 else:
-                    w = _validate_where(w)
-                    where[idx] = w
+                    where[idx] = _validate_where(w)
             _where = " & ".join([f"({w})" for w in com.flatten(where)])
         else:
             # _validate_where ensures we otherwise have a string

pandas/core/dtypes/dtypes.py

@@ -1267,9 +1267,11 @@ def __from_arrow__(
             chunks = array.chunks
 
         results = []
-        for arr in chunks:
-            if isinstance(arr, pyarrow.ExtensionArray):
-                arr = arr.storage
+        for arr_chunk in chunks:
+            arr = (
+                arr_chunk.storage if isinstance(arr_chunk, pyarrow.ExtensionArray)
+                else arr_chunk
+            )
             left = np.asarray(arr.field("left"), dtype=self.subtype)
             right = np.asarray(arr.field("right"), dtype=self.subtype)
             iarr = IntervalArray.from_arrays(left, right, closed=self.closed)

pandas/core/generic.py

@@ -292,9 +292,9 @@ def _init_mgr(
         """passed a manager and a axes dict"""
         for a, axe in axes.items():
             if axe is not None:
-                axe = ensure_index(axe)
+                axe_w_idx = ensure_index(axe)
                 bm_axis = cls._get_block_manager_axis(a)
-                mgr = mgr.reindex_axis(axe, axis=bm_axis)
+                mgr = mgr.reindex_axis(axe_w_idx, axis=bm_axis)
 
         # make a copy if explicitly requested
         if copy: