.csvをpythonで扱う。【退屈なことをpythonにやらせよう耐久】

2017年11月25日 2018年10月21日

1 全CSVファイルを取得し、先頭の行だけを削除する方法
2 プログラムコードの解説
- 2.1 退屈なことをpythonにやらせように書かれていること

全CSVファイルを取得し、先頭の行だけを削除する方法

import csv, os

csv_files = []
#カレントディレクトリのリストから.csvファイルだけを作る。
for file in os.listdir():
 if file.endswith('.csv'):
 csv_files.append(file)
 
#csv_filesをひとつずつ書き込みモードで開きcsv_readerオブジェクトを作成する。
for file in csv_files:
 read_file = open(file)
 csv_reader = csv.reader(read_file)
 output_file = open(os.path.join('csv', file), 'w', newline='')
 file_writer = csv.writer(output_file)
  for row in csv_reader:
   if csv_reader.line_num > 1: 
   file_writer.writerow(row)
 output_file.close()

プログラムコードの解説

16行目の.line_numはcsvリーダーオブジェクト専用のメソッドです。csvの行ごとに番号をもっています。int型として評価できるようです。したがって、＞1とすることで1行目はとばしていることになります。

csvリーダーオブジェクトとライターオブジェクトが近くにあってなんだか気持ち悪いコードになってしまいました。13行目のopen(os.path.join())の組み合わせはいろいろな場所で使えそうなテクニックです。このコードでは’csv’というフォルダに変数fileと同じ名前のファイルを作るという意味になります。

もう少し綺麗にかけるような気もするコードかもしれません。