f = open("stock_data.txt",mode='r', encoding='UTF-8')
stock_data = {}
headers = f.readline().strip().split(",")
for line in f:
line = line.strip().split(",")
stock_data[line[0]] = line
dic_filter_cols = {'当前价':2,'涨跌幅':4,'换手率':8}
while True:
cmd = input("请输入查询条件>>:").strip()
match_count = 0
match_list = []
check_str = str
for k, v in stock_data.items():
if '<' not in cmd and '>' not in cmd:
for i in v:
if cmd.upper() in i.upper():
print(v)
match_count += 1
else:
if '<' in cmd:
match_list = cmd.split('<')
elif '>' in cmd:
match_list = cmd.split('>')
if match_list[0] in dic_filter_cols:
check_str = v[dic_filter_cols.get(match_list[0])].strip('%') + cmd.strip(match_list[0]).strip('%')
if bool(eval(check_str)) is True:
match_count += 1
print(v)
print(f"匹配到了{match_count}条。")
f.close()
