如何在我的 CSV 文件中打印用户设置的两个日期之间的数据？

Question

I have made a CSV file where it stores a book, its author and the year it was published (this file is made from user input). Now I want to ask the user for a starting year and an ending year, and the program should display all the books from the data set published during that time frame. I'm just not sure how to do this

Here is my code so far:

import csv

amount = int(input("How many records would you like to add: "))
count = 0

with open("Books.csv", "w", newline="") as file:
    writer = csv.writer(file)
    writer.writerow(["", "Book", "Author", "Year released"])

while count < amount:

     book = input("Enter a book: ")
     author = input("Enter it's Author: ")
     year = input("Enter the year it released: ")

    headers = [count, book, author, year]
    with open("Books.csv", "a", newline="") as f:
        writer = csv.writer(f)
        writer.writerow(headers)

    count += 1

Answer 1

you can read a csv using pandas.read_csv .

import pandas as pd
df = pd.read_csv("Books.csv")

Now you can filter records from based on year column

filter = (df['Year released'] > start_date) & (df['Year released'] <= end_date)
filter_df = df.loc[filter]

To select the books:

books = filter_df['Book'].tolist()

Answer 2

Pandas is a very good option as Rajat Mishra said - you could also do it with the built-in CSV module:

books = []

with open("books.csv", "r") as file:
    reader = csv.reader(file)
    for row in reader:
        count, book, author, release_year = row
        print(row)
        if release_year is int:
            if int(release_year) > start_year and int(release_year) < end_year:
                books.append(book)


print(books)

Answer 3

I believe that you are able to sort the column by using the built-in sorted function.

import operator
with open("Books.csv", "r") as f:
    reader = csv.reader(f)
    sortedlist = sorted(reader, key=operator.itemgetter(3), reverse=False)

We send 3 parameters to the sorted function:

• reader parameter is the reader object that belongs to the.csv we want to read.
• key parameter indicates the column that will the csv file be sorted based on.
• reverse parameter tells the way sorting, like False = Ascending, True = Descending.