[Python] 纯文本查看 复制代码
def search_by_code():
"""按代码搜索,同时保留类型筛选"""
code = code_entry.get().strip()
selected_type = type_var.get() # 获取当前选中的类型
if code:
status_var.set(f"搜索代码: {code} (类型: {selected_type if selected_type else '全部'})")
root.update()
data = get_stock_changes(selected_type=selected_type, stock_code=code) # 同时传递类型和代码
else:
search_by_type() # 如果没有输入代码,则按类型搜索
populate_treeview(data)
[Python] 纯文本查看 复制代码
def get_stock_changes(selected_type=None, stock_code=None):
"""获取股票异动数据,支持类型和代码联合筛选"""
# ...(其他代码不变)
if selected_type and stock_code:
# 同时按类型和代码筛选
temp_df = temp_df[
(temp_df["板块"] == selected_type) &
(temp_df["代码"].astype(str).str.zfill(6) == stock_code.zfill(6))
]
elif selected_type:
# 仅按类型筛选
temp_df = temp_df[temp_df["板块"] == selected_type]
elif stock_code:
# 仅按代码筛选
temp_df = temp_df[temp_df["代码"].astype(str).str.zfill(6) == stock_code.zfill(6)]
# ...(其他代码不变)