Skip to content

Commit 7b077ac

Browse files
committed
1 parent 09a46a4 commit 7b077ac

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pandas/_libs/algos.pyx

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1050,15 +1050,15 @@ def rank_2d(rank_t[:, :] in_arr, axis=0, ties_method='average',
10501050
if rank_t is object:
10511051
nan_value = Infinity()
10521052
elif rank_t is float64_t:
1053-
nan_value = np.inf
1053+
nan_value = NaN
10541054
elif rank_t is int64_t:
10551055
nan_value = np.iinfo(np.int64).max
10561056

10571057
else:
10581058
if rank_t is object:
10591059
nan_value = NegInfinity()
10601060
elif rank_t is float64_t:
1061-
nan_value = -np.inf
1061+
nan_value = NaN
10621062
elif rank_t is int64_t:
10631063
nan_value = NPY_NAT
10641064

@@ -1120,7 +1120,7 @@ def rank_2d(rank_t[:, :] in_arr, axis=0, ties_method='average',
11201120
if rank_t is object:
11211121
skip_condition = (val is nan_value) and keep_na
11221122
else:
1123-
skip_condition = (val == nan_value) and keep_na
1123+
skip_condition = (val == nan_value or (np.isnan(val) and np.isnan(nan_value))) and keep_na
11241124
if skip_condition:
11251125
ranks[i, argsorted[i, j]] = NaN
11261126

0 commit comments

Comments
 (0)