倒数第二天

博主分享了2021年12月31日倒数第二天的心情,并详细规划了即将到来的元旦假期安排。1号打算在家休息,2号则视情况可能远程值班或逛商场,3号的计划尚未确定。

今天是2021年12月30日,倒数第二天了,大家都在憧憬着美好的元旦假期生活了。

我的元旦假期怎么安排呢,1号准备在家里休息,好好养养精神。

2号看情况吧,也许远程值班也许去商场逛逛。

3号呢,还没有计划号,到时候看看。

下面是我新的代码: import os import pandas as pd import numpy as np from datetime import datetime, timedelta def analyze_stock_data(csv_file, target_date): """ 分析单个股票CSV文件,检查是否满足筛选条件 target_date: 格式为'2025/7/17'的目标日期 """ try: # 读取CSV文件 df = pd.read_csv(csv_file) # 确保数据按日期排序(从旧到新) df['date'] = pd.to_datetime(df['date']) df = df.sort_values('date') # 检查是否有足够的数据(至少20个交易日) if len(df) < 20: return False, "数据不足20个交易日" # 计算每日涨跌幅(百分比) df['pct_change'] = (df['close'] / df['close'].shift(1) - 1) * 100 df = df.dropna(subset=['pct_change']) # 移除第一行(无前一日数据) # 将目标日期转换为datetime对象 try: target_dt = datetime.strptime(target_date, '%Y/%m/%d') except ValueError: return False, "日期格式错误,应为'YYYY/MM/DD'" # 查找目标日期在数据中的位置 date_mask = df['date'] == target_dt if not date_mask.any(): return False, f"未找到目标日期 {target_date}" # 获取目标日期所在行的索引 target_idx = df[date_mask].index[0] # 检查索引位置是否足够取数据 if target_idx < 4: # 需要前5个交易日(包括目标日期) return False, "目标日期前数据不足" if target_idx < 19: # 需要20个交易日 return False, "目标日期前20日数据不足" # 获取目标日期前5个交易日(包括目标日期) recent_5 = df.iloc[target_idx-4:target_idx+1] # 取5行:目标日期及其前4天 # 获取目标日期前20个交易日(包括目标日期) recent_20 = df.iloc[target_idx-19:target_idx+1] # 取20行 # 条件1: 最近20日日均涨幅大于0.3% avg_daily_gain = recent_20['pct_change'].mean() if avg_daily_gain <= -0.3: return False, f"日均涨幅不足(仅{avg_daily_gain:.2f}%)" # 获取最近5个交易日的数据 pct_changes = recent_5['pct_change'].values # 条件2: 近5日有一个交易日涨幅超过9.95% big_gain_days = sum(1 for change in pct_changes if change > 9.95) if big_gain_days < 1: return False, f"大涨日不足(仅{big_gain_days}天)" # 条件3: 近5日有两个交易日上涨 up_days = sum(1 for change in pct_changes if change > 0) if up_days < 2: return False, f"上涨日不足(仅{up_days}天)" # 条件4: 倒数第二天下跌(目标日期的前一天) if pct_changes[-2] >= 0: # 倒数第二天(索引-2) return False, "倒数第二天未下跌" # 条件5: 最后一天上涨(目标日期当天) if pct_changes[-1] <= 0: # 最后一天(索引-1) return False, "最后一天未上涨" return True, "满足所有条件" except Exception as e: return False, f"处理错误: {str(e)}" def filter_stocks(folder_path, target_date): """ 筛选指定文件夹中满足条件的股票CSV文件 target_date: 格式为'2025/7/17'的目标日期 """ # 获取所有CSV文件 csv_files = [f for f in os.listdir(folder_path) if f.endswith('.csv') and f.startswith('sz')] print(f"在 '{folder_path}' 中找到 {len(csv_files)} 个股票文件,开始筛选...") print(f"目标日期: {target_date}") qualified_stocks = [] total_files = len(csv_files) for i, filename in enumerate(csv_files): file_path = os.path.join(folder_path, filename) stock_code = filename.split('.')[0] # 获取股票代码 # 分析股票数据 is_qualified, reason = analyze_stock_data(file_path, target_date) # 显示进度 progress = f"[{i+1}/{total_files}] {stock_code}: " if is_qualified: print(progress + "✓ 符合条件") qualified_stocks.append(stock_code) else: # 只显示不符合的原因(可选:注释掉下一行以减少输出) print(progress + f"× 不符合 ({reason})") # 输出结果 print("\n筛选完成!") print(f"符合条件股票数量: {len(qualified_stocks)}/{total_files}") if qualified_stocks: print("\n符合条件的股票代码:") for stock in qualified_stocks: print(stock) # 保存结果到文本文件 result_txt = os.path.join(folder_path, f"qualified_stocks_{target_date.replace('/', '')}.txt") with open(result_txt, 'w', encoding='utf-8') as f: f.write("\n".join(qualified_stocks)) print(f"\n文本结果已保存到: {result_txt}") # 创建表格并保存到指定位置 save_table(qualified_stocks, target_date) return qualified_stocks def save_table(stock_codes, target_date): """ 将筛选结果保存为表格文件 stock_codes: 符合条件的股票代码列表 target_date: 目标日期 """ # 创建DataFrame result_df = pd.DataFrame({ "股票代码": stock_codes, "筛选日期": target_date, "筛选条件": "回首望月策略" }) # 设置保存路径 save_dir = r"D:\股票量化数据库" if not os.path.exists(save_dir): os.makedirs(save_dir) # 保存为CSV文件 csv_path = os.path.join(save_dir, "深证/回首望月.csv") result_df.to_csv(csv_path, index=False, encoding='utf-8-sig') # 保存为Excel文件(可选) excel_path = os.path.join(save_dir, "深证/回首望月.xlsx") result_df.to_excel(excel_path, index=False) print(f"\n表格文件已保存到:") print(f"CSV格式: {csv_path}") print(f"Excel格式: {excel_path}") # 主程序 if __name__ == "__main__": folder_path = r"D:\股票量化数据库\股票csv数据\深证" target_date = "2025/7/17" # 可修改为目标日期 # 检查路径是否存在 if not os.path.exists(folder_path): print(f"错误: 路径 '{folder_path}' 不存在!") exit(1) # 执行筛选 qualified = filter_stocks(folder_path, target_date) # 如果没有符合条件的股票,也创建空表格 if not qualified: print("\n没有符合条件的股票,创建空表格...") save_table([], target_date) 请帮我加一个条件,要求最后一天不能涨停
07-18
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值