从文件获取和读取数据时出错

Question

I'm trying to create a program that reads information about cinema times from a .csv file, where the user can search for movie times by the month it is shown, the company producing the movie and the name of the movie.

The file is formatted like:

Movie name, date , cinema, post code of cinema, company producing, cost

for each line.

The only information that I need to search for are the dates, company producing and the name of the movie.

The first problem I have with this code is, when I search for movies by a company, only the first movie by the company in the file is shown and I want to show all of them.

The second problem I have is that I need to search for movies by month, but the format for date in the file is dd/mm/yy and I am not sure how to search for just the month and ignore the day and year.

If anyone has a solution to either of these problems, I would very much appreciate it and thanks in advance.

Answer 1

For your first problem it seems that your code is correct, I've tested it out and it does indeed print a list where there is a match for company:

>>> company = "one"
>>> moviedata = [("one","two"), ("two","one"), ("two","three")]
>>> print([s for s in moviedata if (company) in s])
[('one', 'two'), ('two', 'one')]

I can only suggest to print out and check if your moviedata is stored in list of tuples.

for second problem just slice the data you have to retrieve the month:

>>> "03/04/2014"[3:5]
'04'

example for your code:

>>> moviedata = [("one","two"), ("two","one"), ("two","three")]
>>> moviedata[1]
('two', 'one')
>>> moviedata[1][1]
'one'
>>> moviedata[1][1][:2]
'on'