Dataframe conversions

CHANGELOG.md

@@ -7,9 +7,12 @@
 #### Functionality improvements
 
 * Convert `by` arg `_by` to allow naming columns `by` in `.mutate()`/`.summarize()`
+* Convert `.to_pandas()`/`.to_polars()` to `.as_pandas()`/`.as_polars()`
 
 #### New functions
 
+* `as_tibble()`
+* `is_tibble()`
 * `where()`
 
 ## v0.2.19

tests/test_tibble.py

@@ -463,3 +463,12 @@ def test_funs_in_a_row():
     df.tail()
     df.arrange('x', 'y')
     assert True, "Functions in a row failed"
+
+def test_conversion():
+    """Can convert"""
+    df = tp.tibble(a = ["a", "a", "a"], b = ["b", "b", "b"], c = range(3))
+    actual = df.as_polars()
+    expected = pl.DataFrame(dict(a = ["a", "a", "a"], b = ["b", "b", "b"], c = range(3)))
+    assert actual.equals(expected), "as_polars failed"
+    assert tp.is_tibble(df) == True, "is_tibble failed"
+    assert tp.as_tibble(actual).equals(df), "as_tibble failed"

tidypolars/tibble_df.py

@@ -15,6 +15,8 @@
 from operator import not_
 
 __all__ = [
+    "as_tibble",
+    "is_tibble",
     "tibble",
     "desc",
     "from_pandas", "from_polars"
@@ -959,6 +961,43 @@ def desc(x):
 class DescCol(pl.Expr):
     pass
 
+def as_tibble(x):
+    """
+    Convert an object to a tibble
+
+    Parameters
+    ----------
+    x : [pl.DataFrame, pd.DataFrame, dict]
+        Object to convert to a tibble
+
+    Examples
+    --------
+    >>> tp.as_tibble(polars_df)
+    """
+    if isinstance(x, pl.DataFrame):
+        out = from_polars(x)
+    elif isinstance(x, dict):
+        out = tibble(x)
+    elif is_tibble(x):
+        out = x
+    else:
+        out = pl.from_dataframe(x)
+    return out
+
+def is_tibble(x):
+    """
+    Is an object to a tibble
+
+    Parameters
+    ----------
+    x : object
+
+    Examples
+    --------
+    >>> tp.is_tibble(df)
+    """
+    return isinstance(x, tibble)
+
 def from_polars(df):
     """
     Convert from polars DataFrame to tibble