Merge pull request #10 from jseabold/gh-9

Support notnull/notna. Closes #9.

Author: jseabold

pandas_selectable/__init__.py

@@ -177,6 +177,14 @@ def isna(self):
     def isnull(self):
         return self._parent.loc[self._parent.index.isnull()]
 
+    @doc(pd.Index.notnull)
+    def notnull(self):
+        return self._parent.loc[self._parent.index.notnull()]
+
+    @doc(pd.Index.notna)
+    def notna(self):
+        return self._parent.loc[self._parent.index.notna()]
+
     @doc(pd.Index.isin)
     def isin(self, values, levels=None):
         idx = self._parent.index.isin(values, levels)
@@ -196,6 +204,8 @@ class SelectableColumn:
     __ge__ = selector_wrapper(pd.Series, "__ge__")
     isna = selector_wrapper(pd.Series, "isna")
     isnull = selector_wrapper(pd.Series, "isnull")
+    notna = selector_wrapper(pd.Series, "notna")
+    notnull = selector_wrapper(pd.Series, "notnull")
     isin = selector_wrapper(pd.Series, "isin")
 
     def __init__(self, parent, series=None):

pandas_selectable/tests/test_select.py

@@ -19,6 +19,8 @@ def test_select(dta):
     assert ((dta.select.A != 3).A != 3).all()
     assert len(dta.select.A.isnull()) == 0
     assert len(dta.select.A.isna()) == 0
+    assert dta.select.A.notnull().shape == (15, 4)
+    assert dta.select.A.notna().shape == (15, 4)
     assert len(dta.select.A.isin([5])) == 1
 
 
@@ -66,6 +68,8 @@ def test_index(dta):
 
     assert len(dta.select.index.isna()) == 0
     assert len(dta.select.index.isnull()) == 0
+    assert dta.select.index.notnull().shape == (15, 4)
+    assert dta.select.index.notna().shape == (15, 4)
     assert len(dta.select.index.isin([10, 11])) == 2