Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Make errobars work with astropy units #10118

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 3 commits into from
Closed

Make errobars work with astropy units #10118

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
a26b3df
Make errobars work with astropy units
dstansby Dec 27, 2017
1dd14e3
Catch type error
dstansby Dec 27, 2017
ceaee97
Check for arrays instead in unit support
dstansby Dec 27, 2017
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 27 additions & 9 deletions lib/matplotlib/axes/_axes.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2965,13 +2965,24 @@ def _bool_asarray_helper(d, expected):

def xywhere(xs, ys, mask):
"""
return xs[mask], ys[mask] where mask is True but xs and
ys are not arrays
Return xs[mask], ys[mask] where mask is True but xs and
ys are not arrays.

If they are arrays, just return xs[mask] and ys[mask].
"""
assert len(xs) == len(ys)
assert len(xs) == len(mask)
xs = [thisx for thisx, b in zip(xs, mask) if b]
ys = [thisy for thisy, b in zip(ys, mask) if b]

if isinstance(xs, np.ndarray):
xs = xs[mask]
else:
xs = [thisx for thisx, b in zip(xs, mask) if b]

if isinstance(ys, np.ndarray):
ys = ys[mask]
else:
ys = [thisy for thisy, b in zip(ys, mask) if b]

return xs, ys

def extract_err(err, data):
Expand Down Expand Up @@ -3005,11 +3016,18 @@ def extract_err(err, data):
if (len(err) != len(data) or np.size(fe) > 1):
raise ValueError("err must be [ scalar | N, Nx1 "
"or 2xN array-like ]")
# using list comps rather than arrays to preserve units
low = [thisx - thiserr for (thisx, thiserr)
in cbook.safezip(data, err)]
high = [thisx + thiserr for (thisx, thiserr)
in cbook.safezip(data, err)]
if isinstance(data, np.ndarray):
low = data - err
else:
# using list comps rather than arrays to preserve units
low = [thisx - thiserr for (thisx, thiserr)
in cbook.safezip(data, err)]
if isinstance(data, np.ndarray):
high = data + err
else:
# using list comps rather than arrays to preserve units
high = [thisx + thiserr for (thisx, thiserr)
in cbook.safezip(data, err)]
return low, high

if xerr is not None:
Expand Down