Commit 47440353 authored by Jan Kiene's avatar Jan Kiene
Browse files

preserve full path for split files

parent 228de042
Loading
Loading
Loading
Loading
+8 −4
Original line number Diff line number Diff line
@@ -46,9 +46,11 @@ def main(csv_files: list):

        mask_vbr = df["conf"].str.contains("RS")
        df_vbr = df[mask_vbr]
        filename_vbr = f"{csv_file.stem}_rate_sw{csv_file.suffix}"
        filename_vbr = (csv_file.with_suffix("") / "_rate_sw").with_suffix(
            csv_file.suffix
        )
        df_vbr.to_csv(
            filename_vbr,
            str(filename_vbr),
            index=False,
            sep=";",
        )
@@ -61,9 +63,11 @@ def main(csv_files: list):
                df_cbr["bitrate"] >= min_br, df_cbr["bitrate"] <= max_br
            )
            df_lvl = df_cbr[mask_lvl]
            filename_lvl = f"{csv_file.stem}_{lvl_suffix}{csv_file.suffix}"
            filename_lvl = (csv_file.with_suffix("") / lvl_suffix).with_suffix(
                csv_file.suffix
            )
            df_lvl.to_csv(
                filename_lvl,
                str(filename_lvl),
                index=False,
                sep=";",
            )