Merge pull request #537 from vanithakattumuri/main

#2 Updated the file CFPGrowthPlus.py.

PAMI/extras/convert/CSV2Parquet.py

@@ -49,7 +49,7 @@ class CSV2Parquet:
         :**Attributes**:    - **getMemoryUSS** (*float*) -- *Returns the memory used by the process in USS.*
                             - **getMemoryRSS** (*float*) -- *Returns the memory used by the process in RSS.*
                             - **getRuntime()** (*float*) -- *Returns the time taken to execute the conversion.*
-                            - **printStats()** -- * Prints statistics about memory usage and runtime.*
+                            - **printStats()** -- *Prints statistics about memory usage and runtime.*
 
         :**Methods**:       - **convert()** -- *Reads the input file, converts it to a Parquet file, and tracks memory usage and runtime.*
 

PAMI/extras/graph/DF2Tex.py

@@ -1,22 +1,18 @@
 # DF2Tex is used to convert the given dataframe into LatexFile.
 #
 # **Importing this algorithm into a python program**
-# --------------------------------------------------------
 #
-#     from PAMI.extras.graph import DF2Tex
+#             from PAMI.extras.graph import DF2Tex
 #
-#     obj = DF2Tex()
+#             obj = DF2Tex()
 #
-#     obj.generateLatexCode(result, "minSup", "runtime", "algorithmColumn")
+#             obj.generateLatexCode(result, "minSup", "runtime", "algorithmColumn")
 #
-#     obj.print()
+#             obj.print()
 #
-#     obj.save("outputFile.tex")
+#             obj.save("outputFile.tex")
 #
 
-
-
-
 __copyright__ = """
 Copyright (C)  2021 Rage Uday Kiran
 
@@ -34,63 +30,101 @@
      along with this program.  If not, see <https://www.gnu.org/licenses/>.
 """
 
-
 import pandas as pd
 
-
 class DF2Tex:
     """
-    :Description:  generateLatexFileFromDatFrame is used to convert the given dataframe into LatexFile.
+    **About this algorithm**
+
+    :**Description**:  generateLatexFileFromDataFrame is used to convert the given DataFrame into a LaTeX file.
+
+    :**Reference**:
+
+    **Attributes:** - **latexCode** (*str*) -- Stores the generated LaTeX code.
+
+    **Methods:**    - **generateLatexCode(result: pd.DataFrame, xColumn, yColumn, algorithmColumn=None)** -- *Generates LaTeX code based on DataFrame columns.*
+
+                    - **print_latex()** -- *Prints the LaTeX code.*
+
+                    - **save(outputFileName: str)** -- *Saves the LaTeX code to a .tex file.*
+
+    **Importing this algorithm into a Python program**
 
-    **Importing this algorithm into a python program**
-    --------------------------------------------------------
     .. code-block:: python
 
             from PAMI.extras.graph import DF2Tex
 
+            result = Dataframe
+
             obj = DF2Tex()
 
             obj.generateLatexCode(result, "minSup", "runtime", "algorithmColumn")
 
             obj.print()
 
             obj.save("outputFile.tex")
-
     """
 
+    def __init__(self):
+        self.latexCode = ""
+
     def generateLatexCode(self, result: pd.DataFrame, xColumn, yColumn, algorithmColumn=None) -> None:
+        """
+        Generate LaTeX code from a given DataFrame.
+        """
+        # Check if DataFrame is empty
+        if result.empty:
+            raise ValueError("The input DataFrame is empty. Please provide a DataFrame with data.")
+
+        # Check if required columns exist
+        requiredColumns = [xColumn, yColumn]
+        if algorithmColumn:
+            requiredColumns.append(algorithmColumn)
+        for col in requiredColumns:
+            if col not in result.columns:
+                raise ValueError(f"Column '{col}' does not exist in the DataFrame.")
+
         titles = [xColumn, yColumn]
-        legendary = pd.unique(result.iloc[:, 0].values.ravel())
+        legendary = pd.unique(result[algorithmColumn] if algorithmColumn else result.iloc[:, 0].values.ravel())
         xaxisValues = result[xColumn].values
         yaxisValues = result[yColumn].values
-        if algorithmColumn is None:
-            algo = result.iloc[:, 0].values
-        else:
-            algo = result[algorithmColumn].values
+        algo = result[algorithmColumn].values if algorithmColumn else result.iloc[:, 0].values
         xLabel = titles[0]
-        color = ['red', 'blue', 'green', 'black', 'yellow']
+        colors = ['red', 'blue', 'green', 'black', 'yellow']
         latexCode = ""
         latexCode += "\\begin{axis}[\n\txlabel={\\Huge{" + xLabel + "}},"
         latexCode += "\n\tylabel={\\Huge{" + titles[1] + "}},"
         latexCode += "\n\txmin=" + str(min(xaxisValues)) + ", xmax=" + str(max(xaxisValues)) + ",]\n"
-        for num in range(0, len(legendary)):
-            latexCode += "\\addplot+  [" + color[num] + "]\n\tcoordinates {\n"
-            for num2 in range(0, len(xaxisValues)):
-                if legendary[num] == algo[num2]:
-                    latexCode += "(" + str(xaxisValues[num2]) + "," + str(yaxisValues[num2]) + ")\n"
-            latexCode += "\t};   \\addlegendentry{" + legendary[num] + "}\n"
-            if num + 1 == len(legendary):
-                latexCode += "\\end{axis}"
-        DF2Tex.latexCode = latexCode
-
-    def print(self):
-        print(DF2Tex.latexCode)
-
-    def save(outputFileName):
-        with open(outputFileName, "w") as latexwriter:
-            latexwriter.write(DF2Tex.latexCode)
-        print(f"Latex file saved as {outputFileName}")
 
+        for num, legend in enumerate(legendary):
+            color = colors[num % len(colors)]  # Ensure color index is within range
+            latexCode += "\\addplot+  [" + color + "]\n\tcoordinates {\n"
+            for x, y, a in zip(xaxisValues, yaxisValues, algo):
+                if legend == a:
+                    latexCode += f"({x},{y})\n"
+            latexCode += "\t};   \\addlegendentry{" + str(legend) + "}\n"
+        latexCode += "\\end{axis}"
+        self.latexCode = latexCode
+
+    def print_latex(self):
+        """
+        Print the generated LaTeX code.
+        """
+        print(self.latexCode)
+
+    def save(self, outputFileName):
+        """
+        Save the generated LaTeX code to a file.
+
+        :param outputFileName: The name of the output .tex file.
+        :raises ValueError: If LaTeX code has not been generated.
+
+        """
+        if not self.latexCode:
+            raise ValueError("LaTeX code is empty. Please generate it before saving.")
+        with open(outputFileName, "w") as latexwriter:
+            latexwriter.write(self.latexCode)
+        print(f"LaTeX file saved as {outputFileName}")
 
 # Example usage
 result = pd.DataFrame()
@@ -101,5 +135,4 @@ def save(outputFileName):
 # printLatexCode function prints the output of latex file
 obj.print()
 # save function gives the outputFile
-obj.save("outputFile.tex")
-
+obj.save("outputFile.tex")