FIX: restore (and test) handling of nan in hist data

tacaswell · tacaswell · commit 503012da09af · 2019-03-31T00:42:51.000-04:00
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -6666,6 +6666,14 @@ def hist(self, x, bins=None, range=None, density=None, weights=None,
                 _w = np.concatenate(w)
             else:
                 _w = None
+            if bin_range is None:
+                xmin = np.inf
+                xmax = -np.inf
+                for xd in x:
+                    xmin = np.min(xmin, np.nanmin(xd))
+                    xmax = np.max(xmax, np.nanmax(xd))
+                bin_range = (xmin, xmax)
+
             bins = histogram_bin_edges(np.concatenate(x),
                                        bins, bin_range, _w)
         else:
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -6329,3 +6329,17 @@ def test_hist_auto_bins():
     _, bins, _ = plt.hist([[1, 2, 3], [3, 4, 5, 6]], bins='auto')
     assert bins[0] <= 1
     assert bins[-1] >= 6
+
+
+def test_hist_nan_data():
+    fig, (ax1, ax2) = plt.subplots(2)
+
+    data = [1, 2, 3]
+    nan_data = data + [np.nan]
+
+    bins, edges, _ = ax1.hist(data)
+    with np.errstate(invalid='ignore'):
+        nanbins, nanedges, _ = ax2.hist(nan_data)
+
+    assert np.allclose(bins, nanbins)
+    assert np.allclose(edges, nanedges)
