Skip to content

Commit 022ef91

Browse files
committed
Get rid of nonzero() in dropna()
1 parent 9b2ce5f commit 022ef91

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pandas/core/frame.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4569,7 +4569,10 @@ def dropna(self, axis=0, how='any', thresh=None, subset=None,
45694569
else:
45704570
raise TypeError('must specify how or thresh')
45714571

4572-
result = self._take(mask.nonzero()[0], axis=axis)
4572+
if axis == 0:
4573+
result = self.loc[mask]
4574+
else:
4575+
result = self.loc[:, mask]
45734576

45744577
if inplace:
45754578
self._update_inplace(result)

0 commit comments

Comments
 (0)