1. import pandas as pd
  3. def export_results(results, filename="results.csv"):
  4. """
  5. Exports experiment results to a CSV file.
  7. Args:
  8. results (dict): A dictionary containing the results. Keys are column names, values are lists of data.
  9. filename (str): The name of the CSV file to create. Defaults to "results.csv".
  10. """
  11. try:
  12. df = pd.DataFrame(results) # Create a Pandas DataFrame
  13. df.to_csv(filename, index=False) # Export DataFrame to CSV, no index
  14. print(f"Results exported to {filename}")
  15. except Exception as e:
  16. print(f"Error exporting results: {e}")
  20. if __name__ == '__main__':
  21. # Example usage:
  22. sample_results = {
  23. "Column1": [1, 2, 3, 4, 5],
  24. "Column2": [6.0, 7.0, 8.0, 9.0, 10.0],
  25. "Column3": ["a", "b", "c", "d", "e"]
  26. }
  27. export_results(sample_results, "my_experiment_results.csv")
  29. #Example of error handling with bad data
  30. bad_results = {
  31. "Column1": [1, 2, "a", 4, 5],
  32. "Column2": [6.0, 7.0, 8.0, 9.0, 10.0],
  33. }
  34. export_results(bad_results, "bad_results.csv")

Add your comment