【Python】CSV内、年月日をカウントしたい

Question

以下のCSVファイル
年月日
2020/1/1
2020/1/1
2020/1/2
2020/1/3
2020/1/3
2020/1/5
2020/1/5
2020/1/5
2020/1/6

を

以下の様に同じ年月日の数を集計し、csv出力したい　※期間は1年間分
年月日,カウント
2020/1/1,2
2020/1/2,1
2020/1/3,2
2020/1/4,0
2020/1/5,3
2020/1/6,1

ご教授の程、宜しくお願い致します。

Answer 1

from collections import Counter

with open("dates.csv") as IN, open("count.csv", "w") as OUT:
    for key, count in sorted(Counter(map(str.strip, IN)).items()):
        print(key, count, sep=",", file=OUT)

追記：以下、ヘッダ対応版

from collections import Counter

with open("dates.csv") as IN, open("count.csv", "w") as OUT:
    next(IN)
    print("年月日,カウント", file=OUT)
    for key, count in sorted(Counter(map(str.strip, IN)).items()):
        print(key, count, sep=",", file=OUT)

追記：存在しない日付

ここまでくると無理やり感が。。。存在しない日付等、高度なことをしたければPandasの方がシンプルに書ける。

from collections import ChainMap
from collections import Counter
from collections import defaultdict
from itertools import chain

with open("dates.csv") as IN, open("count.csv", "w") as OUT:
    next(IN)
    counter = ChainMap(Counter(map(str.strip, IN)), defaultdict(int))
    print("年月日,カウント", file=OUT)
    for key in sorted(set(chain(counter, ["2020/1/4"]))):
        print(key, counter[key], sep=",", file=OUT)

Comments

[lemonesour]

YouheiSakurai様
早々にご教授頂き、ありがとうざいます。大変助かります。
因みに存在しない年月日についても”0”としてカウントしたいのですが、お分かりになりますでしょうか？
お手数お掛け致します。

[lemonesour]

YouheiSakurai様
ご教授ありがとうございました。

Answer 2

collections.Counter() で集計して、出力すればよいです。

from collections import Counter

with open("sample.csv") as f:
    lines = f.read().splitlines()[1:]  # [1:] はヘッダーをスキップするため


counts = Counter(lines)
print(counts)

with open("output.csv", "w") as f:
    f.write("年月日,カウント\n")
    for date, c in counts.items():
        f.write(f"{date},{c}\n")

追記

因みに存在しない年月日についても”0”としてカウントしたい(上記”2020/1/4,0”のように)のですが、お分かりになりますでしょうか？

この条件が入ってくると、結構複雑になるので、pandas に頼ります。

import pandas as pd

df = pd.read_csv("sample.csv", parse_dates=["年月日"])

# 月日の一覧
dates = pd.DataFrame({"年月日": pd.date_range(df["年月日"].min(), df["年月日"].max())})
# カウントの一覧
count = df.groupby("年月日").size().to_frame("カウント")

# 結合する。
out_df = dates.merge(count, how="left", on="年月日").fillna(0).convert_dtypes()
out_df.to_csv("output.csv", index=False)

年月日,カウント
2020-01-01,2
2020-01-02,1
2020-01-03,2
2020-01-04,0
2020-01-05,3
2020-01-06,1

Comments

[lemonesour]

titoi様
早々にご教授頂き、ありがとうざいます。大変助かります。
因みに存在しない年月日についても”0”としてカウントしたい(上記”2020/1/4,0”のように)のですが、お分かりになりますでしょうか？
お手数お掛け致します

[tiitoi]

追記しました

[lemonesour]

ご教授ありがとうございました。