如何从任何开始日期和任何结束日期获取SUM。 (file.txt)Python

时间:2016-06-02 07:40:08

标签: python

您好我已经被困在这几周了。如果我有一个保存在file.dat文件或文本文件中的列表,其中包含以下信息:

1-3-2014,job4,51081.5,10
10-10-2014,job3,1010.1,35
4-2-2015,job2,61824.7,35
2-4-2016,Job1,222.1,12

(用逗号分隔,因为它们在不同的标题下,如下所示,我已经编写了代码,在下面的函数中显示它)

print (" Heading1      Heading 2    heading 3   heading 4 ")

         1-3-2014      job4         $51081.5     10
         10-10-2014    job3         $1010.1      35
         4-2-2015      job2         $61824.7     35
         2-4-2016      jon1         $222.1       12

我想要什么。 用户输入开始日期(我的文件中的开始日期,例如2014年1月3日)

用户输入结束日期(结束日期也来自文件,例如2015年4月2日)

打印总和的程序(例如总数是$ ......(从开始$$到结束日期的总和),它应该添加开始日期的$和所有日期之间的日期一直到结束日期并打印最终总和。

感谢

下面的代码我已经尝试了,但得到的时间数据与格式'%d /%m /%Y'不匹配。 (也可以在一个函数中执行此操作,因为我听说它可能是正确的方法。

来自datetime import datetime

begin_date = datetime.strptime(input('Enter start date (dd mm yyyy): '), '%d %m %Y')
end_date = datetime.strptime(input('Enter end date (dd mm yyyy): '), '%d %m %Y')

with open ('data.txt', 'r') as f:
total = 0
for line in f:
    strlist = line.rstrip().split(',')

    dt = datetime.strptime(strlist[0], '%d-%m-%Y')

    if begin_date <= dt <= end_date:
        total+=float(strlist[2].split('$')[1])




print(total)

1 个答案:

答案 0 :(得分:1)

您好,您可以使用以下代码,

from datetime import datetime

def calculate(file_name, s_date, e_date):
    dollar = 0
    _format = '%d-%m-%Y'
    start_date = datetime.strptime(s_date, _format)
    end_date = datetime.strptime(e_date, _format)
    with open(file_name) as f:
        lines = f.readlines()
        for line in lines:
            l = line.split(',')
            date = datetime.strptime(l[0], _format)
            if date >= start_date and date <= end_date:
                dollar = dollar + float(l[2])
    return dollar

start_date = raw_input('Enter start date')
end_date = raw_input('Enter end date')
result = calculate('file.dat', start_date, end_date)
print result