From 8e178efb1bc722322f24d87da2d965924ace41a5 Mon Sep 17 00:00:00 2001 From: Dawid Makar Date: Wed, 27 Sep 2023 01:34:49 +0200 Subject: [PATCH] Automatic refactoring. Refactoring step id: UUID('3fe98d38-a9f7-4882-87ea-42502d433c61') --- pandas/core/internals/construction.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/pandas/core/internals/construction.py b/pandas/core/internals/construction.py index f080683d76df7..7ce2a20438109 100644 --- a/pandas/core/internals/construction.py +++ b/pandas/core/internals/construction.py @@ -675,7 +675,8 @@ def reorder_arrays( if columns is not None: if not columns.equals(arr_columns): # if they are equal, there is nothing to do - new_arrays: list[ArrayLike] = [] + new_arrays: list[ArrayLike | None] + new_arrays = [None] * len(columns) indexer = arr_columns.get_indexer(columns) for i, k in enumerate(indexer): if k == -1: @@ -684,9 +685,12 @@ def reorder_arrays( arr.fill(np.nan) else: arr = arrays[k] - new_arrays.append(arr) + new_arrays[i] = arr - arrays = new_arrays + # Incompatible types in assignment (expression has type + # "List[Union[ExtensionArray, ndarray[Any, Any], None]]", variable + # has type "List[Union[ExtensionArray, ndarray[Any, Any]]]") + arrays = new_arrays # type: ignore[assignment] arr_columns = columns return arrays, arr_columns @@ -1038,4 +1042,4 @@ def convert(arr): arrays = [convert(arr) for arr in content] - return arrays + return arrays \ No newline at end of file