步步汇盈

标题: 请大神完善跨周期指标 [打印本页]

作者: 临沧海 时间: 2023-11-13 21:09
标题: 请大神完善跨周期指标
下面指标可以准确表达日线cr，但2日、周线的cr不准确，请大神指教：
import pandas as pd
from stockstats import StockDataFrame

def calculate_CR(filepath):
# 读取文件数据
df = pd.read_table(filepath,
                     names=['date', 'open', 'high', 'low', 'close', 'vol', 'value'],
                     index_col=0, sep=' ', parse_dates=['date'])

#获取各周期指标
stock = StockDataFrame.retype(df)  # 使用 StockDataFrame 类处理原始数据 df，将其转换为股票数据类型
CR = stock['cr']  # 获取 CR 指标数据

# 将每日数据转换为特定频率下的数据
d2_data = stock.resample('2D').last()  # 转换为每两天的数据
d2_data.index -= pd.DateOffset(days=-1)  # 调整索引日期，推迟显示1天

weekly_data = stock.resample('W').last()  # 转换为每周数据
weekly_data.index -= pd.DateOffset(days=2)  # 调整索引日期，提前2天出

monthly_data = stock.resample('M').last()  # 转换为每月数据
monthly_data.index = monthly_data.index.to_period('M').to_timestamp('M') + pd.offsets.MonthEnd(1)  # 将索引调整为每月最后一天
daily_data = stock.resample('D').last()  # 转换为每日数据
# 创建一个日期范围，覆盖您的数据日期范围
date_range = pd.date_range(start=daily_data.index.min(), end=daily_data.index.max(), freq='B')
# 创建一个新的DataFrame，索引为交易日日期
df = pd.DataFrame(index=date_range)
# 将每个交易数据列填充到同一频率，并加入到新的DataFrame中
#上面对df进行了重新索引、填充缺失值以及删除空行操作，行情数据变了，需要重新输入
df['open'] = daily_data['open']
df['high'] = daily_data['high']
df['low'] = daily_data['low']
df['close'] = daily_data['close']
df = df.reindex(daily_data.index)  # 重新索引 DataFrame，以保留原始日期顺序并插入缺失的日期

df['CR_1'] = CR
df['CR_2'] = d2_data['cr'].resample('D').last().fillna(method='bfill')  # 使用每两天的数据填充缺失值
df['CR_2'].fillna(df['CR_2'].mean(), inplace=True)  # 使用均值填充 CR_2 列的缺失值
df['CR_5'] = weekly_data['cr'].resample('D').last().fillna(method='bfill')  # 使用每周数据填充缺失值
df['CR_20'] = monthly_data['cr'].resample('D').last().fillna(method='bfill')
# 对新的DataFrame进行重新索引，以保留原始日期顺序并插入缺失的日期
df = df.reindex(pd.date_range(start=df.index.min(), end=df.index.max(), freq='B'))
df = df.dropna(subset=['CR_1'])  # 以 CR_1 列为基准，删除空行
#df.to_excel('output.xlsx') #输出df

return df[['CR_1', 'CR_2', 'CR_5', 'CR_20']]

if __name__ == '__main__':
filepath = 'stockdata.txt'  # 文件路径，请根据实际情况修改
result = calculate_CR(filepath)
print(result)
result.to_excel('output.xlsx') #输出df

欢迎光临步步汇盈 (http://bbs.58851.com/)