需要帮助?

试试AI小助手吧

获取周线函数

用户头像mx_***307okn
2023-10-30 发布

周线函数

周线函数是指将股票的价格数据按照周为单位进行统计和分析的功能。

通过周线函数,投资者可以看到股票在一周内的开盘价、收盘价、最高价和最低价等数据,同时还可以看到股票的成交量和成交额等信息。

这些数据可以帮助投资者更好地了解股票的走势和趋势从而做出更明智的投资决策

代码

# 周线函数
def day_to_week(df):
    df = find_next_Monday(df)
    df['week_dif']=df['week'].diff().shift(-1)
    df = df[df.week_dif!=0]
    return df

def find_next_Monday(df):
    remove = 0
    length = len(df)
    for i in range (0,length):
        if df['weekday'][i] != 1:
            remove = remove +1
        else:
            break
    df = df.tail(length-remove)
    return df

def _attribute_history(stock, num, attr):
    import datetime
    import pandas as pd
    is_same_week = []
    today = get_datetime().strftime("%Y%m%d")
    yesterday = (get_datetime() + datetime.timedelta(-1)).strftime("%Y%m%d")

    h = get_price(stock, bar_count=num*5, end_date=yesterday, fre_step='1d', fields=attr, is_panel = 0)

    for index in h.index:
        is_same_week.append(index.isocalendar())

    df_isw = pd.DataFrame(is_same_week, columns=['year','week','weekday'], index=h.index)
    df = pd.concat([h, df_isw], axis=1)  
    df = day_to_week(df)
    return df

回测

评论