imsciences 0.6.1.5__py3-none-any.whl → 0.6.1.6__py3-none-any.whl

imsciences/datafunctions.py

@@ -209,8 +209,8 @@ class dataprocessing:
 
         print("\n37. Replace substrings in column of strings")
         print("   - Description: Replace substrings in column of strings based off dictionary, can also change column to lower")
-        print("   - Usage: replace_substrings(df, column, replacements, to_lower=False)")
-        print("   - Example: replace_substrings(df, 'Influencer Handle', replacement_dict, to_lower=True)") 
+        print("   - Usage: replace_substrings(df, column, replacements, to_lower=False, new_column=None)")
+        print("   - Example: replace_substrings(df, 'Influencer Handle', replacement_dict, to_lower=True, new_column='Short Version')") 
 
         print("\n38. Add totals column")
         print("   - Description: Sums all columns with the option to exclude an date column to create a total column")
@@ -1371,29 +1371,39 @@ class dataprocessing:
 
         return df
 
-    def replace_substrings(self, df, column, replacements, to_lower=False):
-        """
-        Replaces substrings in a column of a DataFrame based on a dictionary of replacements. 
-        Optionally converts the column values to lowercase.
-
-        Args:
-        df (pd.DataFrame): The DataFrame containing the column to modify.
-        column (str): The column name where the replacements will be made.
-        replacements (dict): A dictionary where keys are substrings to replace and values are the replacement strings.
-        to_lower (bool, optional): If True, the column values will be converted to lowercase before applying replacements. Default is False.
-
-        Returns:
-        pd.DataFrame: The DataFrame with the specified replacements made, and optionally with lowercase strings.
-        """
-        
-        # Apply substring replacements
-        for old, new in replacements.items():
-            df[column] = df[column].str.replace(old, new, regex=False)
+        def replace_substrings(self, df, column, replacements, to_lower=False, new_column=None):
+            """
+            Replaces substrings in a column of a DataFrame based on a dictionary of replacements. 
+            Optionally converts the column values to lowercase and allows creating a new column or modifying the existing one.
+
+            Args:
+            df (pd.DataFrame): The DataFrame containing the column to modify.
+            column (str): The column name where the replacements will be made.
+            replacements (dict): A dictionary where keys are substrings to replace and values are the replacement strings.
+            to_lower (bool, optional): If True, the column values will be converted to lowercase before applying replacements. Default is False.
+            new_column (str, optional): If provided, the replacements will be applied to this new column. If None, the existing column will be modified. Default is None.
+
+            Returns:
+            pd.DataFrame: The DataFrame with the specified replacements made, and optionally with lowercase strings.
+            """
             
-        # Optionally convert to lowercase
-        if to_lower:
-            df[column] = df[column].str.lower()
-        return df
+            if new_column is not None:
+                # Create a new column for replacements
+                df[new_column] = df[column]
+                temp_column = new_column
+            else:
+                # Modify the existing column
+                temp_column = column
+            
+            # Apply substring replacements
+            for old, new in replacements.items():
+                df[temp_column] = df[temp_column].str.replace(old, new, regex=False)
+            
+            # Optionally convert to lowercase
+            if to_lower:
+                df[temp_column] = df[temp_column].str.lower()
+            
+            return df
 
     def add_total_column(self, df, exclude_col=None, total_col_name='Total'):
         """

{imsciences-0.6.1.5.dist-info → imsciences-0.6.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imsciences
-Version: 0.6.1.5
+Version: 0.6.1.6
 Summary: IMS Data Processing Package
 Author: IMS
 Author-email: cam@im-sciences.com

{imsciences-0.6.1.5.dist-info → imsciences-0.6.1.6.dist-info}/RECORD

@@ -2,13 +2,13 @@ dataprocessing/__init__.py,sha256=quSwsLs6IuLoA5Rzi0ZD40xZaQudwDteF7_ai9JfTPk,32
 dataprocessing/data-processing-functions.py,sha256=vE1vsZ8xOSbR9Bwlp9SWXwEHXQ0nFydwGkvzHXf2f1Y,41
 dataprocessing/datafunctions.py,sha256=vE1vsZ8xOSbR9Bwlp9SWXwEHXQ0nFydwGkvzHXf2f1Y,41
 imsciences/__init__.py,sha256=GIPbLmWc06sVcOySWwNvMNUr6XGOHqPLryFIWgtpHh8,78
-imsciences/datafunctions.py,sha256=Hhx0YgZumQp056xajmsKG-Ejv5hrfMRvnaiC5skRFrA,136526
+imsciences/datafunctions.py,sha256=1DOieL8Xfh6I-5JZlM_XKPwIon-I_VcDjppuvXmhYhk,137236
 imsciences/datapull.py,sha256=TPY0LDgOkcKTBk8OekbD0Grg5x0SomAK2dZ7MuT6X1E,19000
 imsciencesdataprocessing/__init__.py,sha256=quSwsLs6IuLoA5Rzi0ZD40xZaQudwDteF7_ai9JfTPk,32
 imsciencesdataprocessing/datafunctions.py,sha256=vE1vsZ8xOSbR9Bwlp9SWXwEHXQ0nFydwGkvzHXf2f1Y,41
 imsdataprocessing/__init__.py,sha256=quSwsLs6IuLoA5Rzi0ZD40xZaQudwDteF7_ai9JfTPk,32
 imsdataprocessing/datafunctions.py,sha256=vE1vsZ8xOSbR9Bwlp9SWXwEHXQ0nFydwGkvzHXf2f1Y,41
-imsciences-0.6.1.5.dist-info/METADATA,sha256=q0y77puyMCZS6DnLOPD7ZStlZU7ib1k76lGP3H1Gv9c,854
-imsciences-0.6.1.5.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-imsciences-0.6.1.5.dist-info/top_level.txt,sha256=hsENS-AlDVRh8tQJ6-426iUQlla9bPcGc0-UlFF0_iU,11
-imsciences-0.6.1.5.dist-info/RECORD,,
+imsciences-0.6.1.6.dist-info/METADATA,sha256=SbdVxObVs6UW90bJ0eQIAbX1rd0urpG6sNmcxiB5uLw,854
+imsciences-0.6.1.6.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+imsciences-0.6.1.6.dist-info/top_level.txt,sha256=hsENS-AlDVRh8tQJ6-426iUQlla9bPcGc0-UlFF0_iU,11
+imsciences-0.6.1.6.dist-info/RECORD,,