ヘッダー行を抽出 (#2)

c-3lab · Feb 17, 2024 · 7e54c54 · 7e54c54
1 parent 9e47aaf
commit 7e54c54
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -18,8 +18,12 @@
         j += 1
         df = df.replace('\n', '', regex=True).replace('\r', '', regex=True).replace('\r\n', '', regex=True).replace('\n\r', '', regex=True)
         print(df)
+        if j == 1:
+            df_header = df.iloc[:2]
+            print(df_header)
         df.to_csv(f"./files/{j}.csv", index=None)
     integrate("./files", f"./output_files/output{i}.csv")
     shutil.rmtree("./files")
     os.mkdir("./files")
+
 shutil.rmtree("./files")