python - 根据用户输入从两列 csv 文件中检索数据

Question

我有这个名为 sample.csv 的 csv 文件：

CODE     HOME_TEAM     AWAY_TEAM 
0001        Lakers          Jazz          
0002       Raptors        Lakers             
0003      Warriors      Portland           
0004     Mavericks        Lakers          
0005       Celtics     Mavericks

我希望根据用户输入打印团队参与的每场比赛。假设在上面的例子中，用户输入“Lakers”，程序应该打印：

CODE     HOME_TEAM     AWAY_TEAM 
0001        Lakers          Jazz          
0002       Raptors        Lakers                     
0004     Mavericks        Lakers          
    

到目前为止，我编写的代码（见下文）分别打印了这些信息。

import pandas as pd

df = pd.read_csv("sample.csv")

select_Team = input("Which team do you like to see: ")
df2 = df[df["HOME_TEAM"] == select_Team]
df3 = df[df["AWAY_TEAM"] == select_Team]
print(df2)
print(df3)

我应该怎么做才能将信息呈现为一个（就像我上面显示的那样）？

标签： pythonpandascsv

Answer 1

如果你想同时包含这两个条件

df2 = df[(df["HOME_TEAM"] == select_Team) | (df["AWAY_TEAM"] == select_Team)]