在csv（python）的单元格中显示具有关键字的特定行

Question

我有一个CSV文件，它是费用报告，例如：

day       amount    category
monday    1.00      food
tuesday   2.00      entertainment
wednesday 3.00      drinks
...       ...       ...

我想编写一个代码，要求用户选择一个类别，然后显示用户选择的所有信息，即：

用户选择食物：

Day     Amount   Category
Monday  1.00     food
Friday  3.50     food

到目前为止，我有：

category = input("Input the category you wish to view (Beverages, Food, Travel, Entertainment, Clothing): ")
    if not category in expense_report:
        print("Please enter a valid category")

但不确定下一步该怎么做以过滤出用户选择的内容。

Answer 1

假设您的csv文件的名称是data，则可以使用：
print(data[data.Category == category])

将显示csv文件的所有行和列，其中“类别”与用户输入的类别匹配。

我认为使用诸如pandas之类的库会使其效率更高。