Python Removing the Header from CSV Files

投稿者: | 2017年3月15日
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# -Removes the header from all CSV files in the current working directory.

import csv
import os


def main():
    os.makedirs('headerRemoved', exist_ok=True)

    # Loop through every file in the current working directory
    for csvfilemane in os.listdir('.'):
        if not csvfilemane.endswith('.csv'):
            continue
        print(f'Removing header from {csvfilemane} ...')

        # Read the CSV file in (skipping first row).
        csvRows = []
        with  open(csvfilemane) as csvFileobj:
            readerobj = csv.reader(csvFileobj)
            for row in readerobj:
                if readerobj.line_num == 1:
                    continue
                csvRows.append(row)

        # Write out the CSV file.
        with open(os.path.join('headerRemoved', csvfilemane),
                               'w', newline='') as csvFileobj:
            csvWriter = csv.writer(csvFileobj)
            for row in csvRows:
                csvWriter.writerow(row)
        

if __name__ == '__main__':
    main()

カレントディレクトリの.csvファイルのヘッダーを消すスクリプト。

もともとのファイルは変更されず、headerRemovedのフォルダに変更したファイルが保存される。

 

 

Pocket

コメントを残す

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です

CAPTCHA