FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

Merge pull request #11298 from eric-wieser/masked_values-shrink · numpy/numpy@8525d3e · GitHub

/ numpy Public

Commit 8525d3e

Browse files
authored
Merge pull request #11298 from eric-wieser/masked_values-shrink
BUG: Undo behavior change in ma.masked_values(shrink=True)
2 parents 3bb11d6 + 51dc6f4 commit 8525d3e

2 files changed

Lines changed: 10 additions & 2 deletions

File tree

‎numpy/ma/core.py‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2319,8 +2319,10 @@ def masked_values(x, value, rtol=1e-5, atol=1e-8, copy=True, shrink=True):
23192319
mask = np.isclose(xnew, value, atol=atol, rtol=rtol)
23202320
else:
23212321
mask = umath.equal(xnew, value)
2322-
return masked_array(
2323-
xnew, mask=mask, copy=copy, fill_value=value, shrink=shrink)
2322+
ret = masked_array(xnew, mask=mask, copy=copy, fill_value=value)
2323+
if shrink:
2324+
ret.shrink_mask()
2325+
return ret
23242326

23252327

23262328
def masked_invalid(a, copy=True):

‎numpy/ma/tests/test_core.py‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4904,6 +4904,12 @@ def test_masked_values(self):
49044904
res = np.ma.masked_values(np.inf, -np.inf)
49054905
assert_equal(res.mask, False)
49064906

4907+
res = np.ma.masked_values([1, 2, 3, 4], 5, shrink=True)
4908+
assert_(res.mask is np.ma.nomask)
4909+
4910+
res = np.ma.masked_values([1, 2, 3, 4], 5, shrink=False)
4911+
assert_equal(res.mask, [False] * 4)
4912+
49074913

49084914
def test_masked_array():
49094915
a = np.ma.array([0, 1, 2, 3], mask=[0, 0, 1, 0])

0 commit comments

Comments
 (0)

Back | FazBrowse Home | New Git URL