显示 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文件的所有行和列，其中类别与用户输入的类别匹配。

我认为使用像熊猫这样的库会使其更有效率。