``` import baostock as bs import pandas as pd import numpy as np from concurrent.futures import ThreadPoolExecutor from ta.momentum import RSIIndicator import socket # 新增导入标准库socket def get_all_a_shares(): """获取沪深A股列表（无ST）""" try: rs = bs.query_all_stock(day=pd.Timestamp.now().strftime('%Y-%m-%d')) df = pd.DataFrame(rs.data, columns=rs.fields) is_a_share = df['code'].str.match(r'^(sh\.60|sz\.00|sz\.300)') no_st = ~df['code_name'].str.contains('ST') return df[is_a_share & no_st][['code', 'code_name']].values.tolist() except Exception as e: print(f"获取股票列表失败: {str(e)}") return [] def get_rsi(stock): """带连接健康检查的RSI计算""" code, name = stock for _ in range(3): # 最大重试次数 try: # 获取数据 end_date = pd.Timestamp.now().strftime('%Y-%m-%d') start_date = (pd.Timestamp.now() - pd.Timedelta(days=60)).strftime('%Y-%m-%d') rs = bs.query_history_k_data( code, "date,close", start_date=start_date, end_date=end_date, frequency="d", adjustflag="2" ) if rs.error_code != '0': continue df = pd.DataFrame(rs.data, columns=rs.fields) if len(df) < 30: return None df['close'] = pd.to_numeric(df['close']) rsi = RSIIndicator(df['close'], 14).rsi().iloc[-1] return (code, name, rsi) if not np.isnan(rsi) else None # 修改点：使用标准socket异常替代bs.SocketError except (socket.error, ConnectionResetError) as e: print(f"连接异常({e})，正在重试... ({code})") bs.logout() bs.login() # 重连机制 continue except Exception as e: print(f"未知错误: {str(e)}") continue return None def main(): try: # 带验证的登录 login_result = bs.login() if login_result.error_code != '0': print(f"登录失败: {login_result.error_msg}") return stocks = get_all_a_shares() print(f"获取到{len(stocks)}只股票") # 多线程计算RSI with ThreadPoolExecutor(max_workers=5) as executor: results = list(executor.map(get_rsi, stocks[:50])) # 示例取前50只 # 过滤有效结果 valid_results = [r for r in results if r is not None] print(f"成功计算{len(valid_results)}个RSI指标") except Exception as e: print(f"主程序错误: {str(e)}") finally: bs.logout() if __name__ == "__main__": main()```未知错误: module 'baostock' has no attribute 'query_history_k_data' 成功计算0个RSI指标