Python：如何从 CSV 获取工作日？

Question

我有一个包含各种列的 CSV 样本，我必须从给定的 date 列中提取工作日。示例如下：

Name,Birthdate,Age,Address
ABC,3-10-2016 11:00:00AM,21,XYZ Street 21 zone
BCD,3-11-2016 15:54:00PM,22,WXY Street 21/A, S zone
CDW,4-11-2015 21:09:00PM,22,ZYX Street 21Avenue, North Zone

我想读取 CSV 并提取日期以确定给定日期列的工作日

到目前为止，我已经创建了一个代码来读取 CSV 并获取工作日，但我无法在任何其他 CSV 上实现它。

代码如下：

import csv
from datetime import datetime as dt
with open('date.csv', 'r') as f:
  f.readline()

  for line in f:
    date = dt.strptime(line.strip(), '%m-%d-%Y %H:%M:%S').strftime('%A')
    print date

请帮助我，因为这是我学术研究的一部分。

注意：如果问题不清楚，请告诉我。 :)

Answer 1

使用csv module 读取CSV 文件，然后解析一列。由于您有一个带有标题的文件，因此在这里使用DictReader() 方法是最简单的：

import csv
from datetime import datetime


with open(filename, 'rb') as infile:
    reader = csv.DictReader(infile)
    for row in reader:
         birthdate = row['Birthdate']  # keys are named in the first row of your CSV
         birthdate = datetime.strptime(birthdate, '%m-%d-%Y %H:%M:%S')
         print birthdate.strftime('%A')

Answer 2

使用您导入的csv 库，如下所示：

from datetime import datetime as dt        
import csv

with open('date.csv', 'rb') as f_input:
    csv_input = csv.reader(f_input)
    header = next(csv_input)

    for row in csv_input:
        print dt.strptime(row[1], '%m-%d-%Y %H:%M:%S%p').strftime('%A')

这将打印：

Thursday
Friday
Saturday           

这会自动读取 csv 文件中的每一行作为单元格列表。因此，在这种情况下，您使用 row[1] 访问第二列并将其传递给您的 strptime 函数以获取星期几。

Answer 3

试试这个：

import re
from datetime import datetime as dt

test_str = open('date.csv').read()
p = re.compile(ur'\w+,(\d+-\d+-\d+\s\d+:\d+:\d+\w{2})')
m = re.findall(p, test_str)
for i in m:
    print dt.strptime(i, '%m-%d-%Y %H:%M:%S%p').strftime('%A')

输入：

Name,Birthdate,Age,Address
ABC,3-10-2016 11:00:00AM,21,XYZ Street 21 zone
BCD,3-11-2016 15:54:00PM,22,WXY Street 21/A, S zone
CDW,4-11-2015 21:09:00PM,22,ZYX Street 21Avenue, North Zone

输出：

Thursday
Friday
Saturday