df.to_csv(timestamps_path, columns=df_columns, index=True)

BirdVox · Dec 5, 2018 · 48eb4d7 · 48eb4d7
1 parent d9edbfd
commit 48eb4d7
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/birdvoxdetect/core.py b/birdvoxdetect/core.py
@@ -100,11 +100,12 @@ def process_file(
             filepath, suffix + "timestamps.csv", output_dir=output_dir)
         event_times = []
         event_confidences = []
+        df_columns = ["Time (s)", "Confidence (%)"]
         df = pd.DataFrame({
             "Time (s)": event_times,
             "Confidence (%)": event_confidences
         })
-        df.to_csv(timestamps_path, index=True)
+        df.to_csv(timestamps_path, columns=df_columns, index=True)
 
     # Create directory of output clips.
     if export_clips:
@@ -180,7 +181,7 @@ def process_file(
             "Time (s)": event_times,
             "Confidence (%)": event_confidences
         })
-        df.to_csv(timestamps_path, index=True)
+        df.to_csv(timestamps_path, columns=df_columns, index=True)
 
         if export_clips:
             for t in th_peak_timestamps:
@@ -251,7 +252,7 @@ def process_file(
             "Time (s)": event_times,
             "Confidence (%)": event_confidences
         })
-        df.to_csv(timestamps_path, index=True)
+        df.to_csv(timestamps_path, columns=df_columns, index=True)
 
         # Export clips.
         if export_clips:
@@ -309,7 +310,7 @@ def process_file(
             "Time (s)": event_times,
             "Confidence (%)": event_confidences
         })
-        df.to_csv(timestamps_path, index=True)
+        df.to_csv(timestamps_path, columns=df_columns, index=True)
 
         # Export clips.
         if export_clips: