使用second_cars_info.csv制作一个查询引擎,可以通过输入指定的参数获取对应条目的结果。
“该回答引用ChatGPT”
可参考下面的方法,还请测试代码的可行性,如果可行还请 点击采纳,感谢支持!
import pandas as pd
# 读取 csv 文件
df = pd.read_csv("file.csv")
# 获取指定列
def query_result(df, brand=None, name=None, boarding_time=None, km=None, discharge=None, sec_price=None):
query_df = df
if brand is not None:
query_df = query_df[query_df['Brand'] == brand]
if name is not None:
query_df = query_df[query_df['Name'] == name]
if boarding_time is not None:
query_df = query_df[query_df['Boarding time'] == boarding_time]
if km is not None:
query_df = query_df[query_df['Km'] == km]
if discharge is not None:
query_df = query_df[query_df['Discharge'] == discharge]
if sec_price is not None:
query_df = query_df[query_df['Sec price'] == sec_price]
return query_df
# 调用函数查询
result = query_result(df, brand='BrandA', name='NameB')
print(result)