1. import pandas as pd
  2. import os
  4. def export_string_results(results, output_dir="output", dry_run=True):
  5. """
  6. Exports string results to a CSV file for exploratory analysis.
  8. Args:
  9. results (dict): A dictionary where keys are labels and values are lists of strings.
  10. output_dir (str): The directory to save the CSV file. Defaults to "output".
  11. dry_run (bool): If True, only prints the file path without creating the file. Defaults to True.
  12. """
  13. if not os.path.exists(output_dir):
  14. os.makedirs(output_dir)
  16. filename = os.path.join(output_dir, "string_results.csv")
  18. if dry_run:
  19. print(f"Dry run: Would export to {filename}")
  20. else:
  21. df = pd.DataFrame(results)
  22. df.to_csv(filename, index=False)
  23. print(f"Exported to {filename}")
  26. if __name__ == '__main__':
  27. # Example usage
  28. example_results = {
  29. "label_a": ["apple", "banana", "cherry"],
  30. "label_b": ["red", "green", "blue"],
  31. "label_c": ["one", "two", "three"]
  32. }
  34. export_string_results(example_results, output_dir="my_results", dry_run=True)
  35. export_string_results(example_results, output_dir="my_results", dry_run=False)

Add your comment