python - 从 CSV 文件中，计算一行中的唯一值并使用 python 打印总数

Question

我是 Python 新手，我有一个如下所示的 CSV 文件:

Date, Profit
02/2019 , 100
03/2019 , 410
03/2019 , 300
04/2019 , 200

我需要用 Python 编写一段代码来打印文件中有多少个唯一日期

这是我到目前为止所拥有的，但它只打印#1

import os
import csv

with open('budget_data.csv') as file:
    reader = csv.reader(file,delimiter=',')

    t_dates = [1]
    for row in reader:
        if row[0] not in row:
            t_dates.append(row[0])

    print(len(t_dates))

Answer 1

您可以使用 DictReader 将 csv 标题作为键映射到项目。然后，您可以使用集合理解来获取唯一日期的列表并计算它们的长度。

import csv

with open('test.dat') as file:
    reader = csv.DictReader(file, delimiter=',')
    print(len({row['Date'] for row in reader}))