| FazBrowse GitHub Viewer | Trending | | Home |
| Tools: [Download Repo ZIP] [Original HTTPS Page] |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -2319,8 +2319,10 @@ def masked_values(x, value, rtol=1e-5, atol=1e-8, copy=True, shrink=True): | |||
| 2319 | 2319 | mask = np.isclose(xnew, value, atol=atol, rtol=rtol) | |
| 2320 | 2320 | else: | |
| 2321 | 2321 | 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 | ||
| 2324 | 2326 | ||
| 2325 | 2327 | ||
| 2326 | 2328 | def masked_invalid(a, copy=True): | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -4904,6 +4904,12 @@ def test_masked_values(self): | |||
| 4904 | 4904 | res = np.ma.masked_values(np.inf, -np.inf) | |
| 4905 | 4905 | assert_equal(res.mask, False) | |
| 4906 | 4906 | ||
| 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 | + | ||
| 4907 | 4913 | ||
| 4908 | 4914 | def test_masked_array(): | |
| 4909 | 4915 | a = np.ma.array([0, 1, 2, 3], mask=[0, 0, 1, 0]) | |
| Back | FazBrowse Home | New Git URL |
0 commit comments