fixing pyarrow errors in control flow logic

Author: akkik04

pandas/core/construction.py

@@ -566,6 +566,14 @@ def sanitize_array(
     # extract ndarray or ExtensionArray, ensure we have no NumpyExtensionArray
     data = extract_array(data, extract_numpy=True, extract_range=True)
 
+    # GH#61026: when 2D input is allowed (e.g. DataFrame column assignment),
+    # treat a (n, 1) numpy array as a 1D array of length n so downstream code
+    # (including pyarrow-backed StringArray) always sees 1D.
+    if allow_2d and isinstance(data, np.ndarray) and data.ndim == 2:
+        rows, cols = data.shape
+        if cols == 1:
+            data = data[:, 0]
+
     if isinstance(data, np.ndarray) and data.ndim == 0:
         if dtype is None:
             dtype = data.dtype
@@ -612,7 +620,7 @@ def sanitize_array(
 
         if dtype is None:
             # GH#61026: special-case 2D+ object ndarrays when dtype is None.
-            if data.dtype == object and data.ndim > 1:
+            if allow_2d and data.dtype == object and data.ndim > 1:
                 if data.ndim == 2 and data.shape[1] == 1:
                     # allow assigning a (n, 1) object array to a single column.
                     data = data[:, 0]