Added Pandas Horizontal Concat and Pandas Vertical Concat nodes.

2025-02-18 01:36:09 -08:00
parent d6955792b6
commit 754a9705d5
7 changed files with 99 additions and 2 deletions
@@ -0,0 +1,42 @@
+from typing import Any, Dict
+import pandas as pd
+
+
+class PandasHorizontalConcat:
+    """
+    PandasHorizontalConcat:
+        A class for horizontally concatenating two pandas DataFrames.
+    """
+    
+    @classmethod
+    def INPUT_TYPES(cls) -> Dict[str, Any]:
+        """
+        Defines the input types for the function.
+
+        Returns:
+            Dict[str, Any]: A dictionary specifying required input types.
+        """
+        return {
+            "required": {
+                "a_dataframe": ("DATAFRAME", {}),
+                "b_dataframe": ("DATAFRAME", {})
+            }
+        }
+
+    RETURN_TYPES: tuple = ("DATAFRAME",)
+    FUNCTION: str = "f"
+    CATEGORY: str = "Data Analysis"
+
+    def f(self, a_dataframe: pd.DataFrame, b_dataframe: pd.DataFrame) -> tuple:
+        """
+        Horizontally concatenate two pandas DataFrames.
+
+        Args:
+            a_dataframe (DataFrame): The first DataFrame.
+            b_dataframe (DataFrame): The second DataFrame.
+
+        Returns:
+            tuple: A tuple containing the concatenated DataFrame.
+        """
+        df_out = pd.concat([a_dataframe, b_dataframe], axis=1)
+        return (df_out,)
@@ -0,0 +1,42 @@
+from typing import Any, Dict
+import pandas as pd
+
+
+class PandasVerticalConcat:
+    """
+    PandasVerticalConcat:
+        A class for vertically concatenating two pandas DataFrames.
+    """
+    
+    @classmethod
+    def INPUT_TYPES(cls) -> Dict[str, Any]:
+        """
+        Defines the input types for the function.
+
+        Returns:
+            Dict[str, Any]: A dictionary specifying required input types.
+        """
+        return {
+            "required": {
+                "a_dataframe": ("DATAFRAME", {}),
+                "b_dataframe": ("DATAFRAME", {})
+            }
+        }
+
+    RETURN_TYPES: tuple = ("DATAFRAME",)
+    FUNCTION: str = "f"
+    CATEGORY: str = "Data Analysis"
+
+    def f(self, a_dataframe: pd.DataFrame, b_dataframe: pd.DataFrame) -> tuple:
+        """
+        Vertically concatenate two pandas DataFrames.
+
+        Args:
+            a_dataframe (DataFrame): The first DataFrame.
+            b_dataframe (DataFrame): The second DataFrame.
+
+        Returns:
+            tuple: A tuple containing the concatenated DataFrame.
+        """
+        df_out = pd.concat([a_dataframe, b_dataframe], axis=0, ignore_index=True)
+        return (df_out,)