no iteration needed

Author: vangj

docs/source/conf.py

@@ -22,7 +22,7 @@
 author = 'Jee Vang'
 
 # The full version, including alpha/beta/rc tags
-release = '3.0.4'
+release = '3.0.5'
 
 
 # -- General configuration ---------------------------------------------------

pypair/contingency.py

@@ -1997,18 +1997,16 @@ def add_count(x, y):
 
     @staticmethod
     def _to_categorical_counts(a, b, a_vals=None, b_vals=None):
-        df = pd.DataFrame([(x, y) for x, y in zip(a, b)], columns=['a', 'b'])
+        df = pd.DataFrame({'a': a, 'b': b})
 
         a_values = []
         b_values = []
 
         if a_vals is None:
-            a_values = list(df.a.unique())
-            a_values = sorted([v for v in a_values if pd.notna(v)])
+            a_values = sorted([v for v in df.a.unique() if pd.notna(v)])
 
         if b_vals is None:
-            b_values = list(df.b.unique())
-            b_values = sorted([v for v in b_values if pd.notna(v)])
+            b_values = sorted([v for v in df.b.unique() if pd.notna(v)])
 
         table = [[df.query(f'a=="{x}" and b=="{y}"').shape[0] + 1 for y in b_values] for x in a_values]
         return table

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name='pypair',
-    version='3.0.4',
+    version='3.0.5',
     author='Jee Vang',
     author_email='vangjee@gmail.com',
     packages=find_packages(exclude=('*.tests', '*.tests.*', 'tests.*', 'tests')),