CSVファイルを二次元ファイルで読み込んだ後、データをソートしたい。<python.

Question

CSVファイルを二次配列元で読み込んだ後、データをソートする方法を知りたいです。

次のようなCSVファイルを二次元配列で読みこみました。
その後、一番左の日時を日付情報として読み取らせて、ソートしたいです。
どのような処理を行えば下期待できる処理を行えますでしょうか。
教えて頂きたく思います。

|日時|開催場所|主催会社|対応部署
|:--|:--:|--:|
|2017/3/4|Tokyo|M銀行|総務部
2017/3/4|Tokyo|S銀行|総務部
2017/3/3|Tokyo|S信託|総務部
2017/3/5|Tokyo|N証券|総務部
2017/3/5|KObe|M銀行|総務部

....7000件以上このようなデータが続きます。

import csv

with open('data.csv') as fp:
  lst = list(csv.reader(fp))

print(lst)

期待したい処理結果
[['2017/3/3', 'Tokyo', 'S信託', '総務部']
['2017/3/3', 'Tokyo', 'S信託', '総務部']
['2017/3/4', 'Tokyo', 'S銀行', '総務部']
['2017/3/5', 'Tokto', 'N証券', '総務部']
['2017/3/5', 'Kobe', 'M銀行', '総務部']
...

['2018/10/31', 'Tokyo', 'M証券', '総務部']]

Answer 1

pandas を使うのが簡単かと思います。

import pandas as pd

df = pd.read_csv('test.csv', parse_dates=['日時'], encoding='shift-jis')
df = df.sort_values('日時')
df.to_csv('sorted_data.csv', index=False)

追記

csv モジュールを使うやり方

import csv
from datetime import datetime as dt

data = []
with open('test.csv', encoding='shift-jis') as f:
    reader = csv.reader(f)
    header = next(reader)
    data = list(reader)

data = sorted(data, key=lambda x: dt.strptime(x[0], '%Y/%m/%d'))
with open('sorted_data.csv', 'w') as f:
    writer = csv.writer(f)
    writer.writerow(header)
    writer.writerows(data)

こちらで動作確認に使用した csv ファイル

入力
test.csv

日時,開催場所,主催会社,対応部署
2017/3/4,Tokyo,M銀行,総務部
2017/3/4,Tokyo,S銀行,総務部
2017/3/3,Tokyo,S信託,総務部
2017/3/5,Tokyo,N証券,総務部
2017/3/5,KObe,M銀行,総務部

出力
sorted_data.csv

日時,開催場所,主催会社,対応部署
2017/3/3,Tokyo,S信託,総務部
2017/3/4,Tokyo,M銀行,総務部
2017/3/4,Tokyo,S銀行,総務部
2017/3/5,Tokyo,N証券,総務部
2017/3/5,KObe,M銀行,総務部

Comments

[moto_i]

回答ありがとうございます。
自分の環境で試しましたが、下記の文字コードエラーがでています。
UnicodeDecodeError: 'utf-8' codec can't decode byte 0x8b in position 1: invalid start byte

[tiitoi]

csv の文字コードはなんでしょうか？

df = pd.read_csv('data.csv', parse_dates=['日時'])
↓
shift-jis であれば、以下のようにエンコーディングを指定してください。
df= pd.read_csv('data.csv', parse_dates=['日時'], encoding='shift_jis')
もしくは Excel 等で作成したファイルの場合 cp932 の可能性もありますので、上記で駄目なら以下も試してください
df= pd.read_csv('data.csv', parse_dates=['日時'], encoding='cp932')

[moto_i]

フォローありがあとうございます。
エンコーディング指定をencoding='shift_jis' , 'cp932'で試しましたが、それぞれエラーが出てしまいました。ちなみにCSVファイルはExcelで作成したあものです。

[tiitoi]

csv ファイルの文字コードがなにかわかりませんか？
わからない場合は「文字コード 調べ方」等で検索してみてください。

[moto_i]

調べました。SJISになっています。
サクラエディタを使って、CSVファイルの文字コード判定を行いました。
エンコーディングをSJISに設定しましたが、プログラムが走りませんでした。

[tiitoi]

shiftjis であれば、encoding='shift-jis' で読み込めるはずなので、ちょっとわからないですね。
一応、csv モジュールを使うやり方も追記してみました。

[moto_i]

ありがとうございます。ソート処理ができました。
要素数が1つ減っているみたいですね。
要素数を減らさずにソートすることは可能でしょうか？
[['2017/3/3', 'Tokyo', 'S信託', '総務部'], ['2017/3/4', 'Tokyo', 'S銀行', '総務部'], ['2017/3/5', 'Tokto', 'N証券', '総務部'], ['2017/3/5', 'Kobe', 'M銀行', '総務部']]

[tiitoi]

> 要素数が1つ減っているみたいですね。

csv ファイルの1行目に「日時,開催場所,主催会社,対応部署」というヘッダーはあるものだと思っていたのですが、ないのでしょうか？
もしそうなら、header = next(reader) と writer.writerow(header) の行は消してください。

[tiitoi]

動作確認に使用した csv ファイルを追記しました。

[moto_i]

ご丁寧に説明頂きありがとうございます。
すみません。ヘッダーはありましたが、消していました。こちらのミスです。
非常に勉強になりました。非常に感謝しています。