Skip to content

Commit 4d31e4a

Browse files
committed
REF: Use range over arange for iterators
1 parent 6fa4eb4 commit 4d31e4a

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

pandas/core/generic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7047,7 +7047,7 @@ def fillna(
70477047
# see test_fillna_dict_inplace_nonunique_columns
70487048
locs = result.columns.get_loc(k)
70497049
if isinstance(locs, slice):
7050-
locs = np.arange(self.shape[1])[locs]
7050+
locs = range(self.shape[1])[locs]
70517051
elif isinstance(locs, np.ndarray) and locs.dtype.kind == "b":
70527052
locs = locs.nonzero()[0]
70537053
elif not (

pandas/core/indexing.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2308,11 +2308,11 @@ def _ensure_iterable_column_indexer(self, column_indexer):
23082308
"""
23092309
Ensure that our column indexer is something that can be iterated over.
23102310
"""
2311-
ilocs: Sequence[int | np.integer] | np.ndarray
2311+
ilocs: Sequence[int | np.integer] | np.ndarray | range
23122312
if is_integer(column_indexer):
23132313
ilocs = [column_indexer]
23142314
elif isinstance(column_indexer, slice):
2315-
ilocs = np.arange(len(self.obj.columns))[column_indexer]
2315+
ilocs = range(len(self.obj.columns))[column_indexer]
23162316
elif (
23172317
isinstance(column_indexer, np.ndarray) and column_indexer.dtype.kind == "b"
23182318
):

0 commit comments

Comments
 (0)