请教大佬,tmp = df.loc[[s]] # 此处为什么要用双层方括号[[ ]]:
# 计算连板数
def get_continue_count_df(hl_list, date, watch_days):
df = pd.DataFrame()
for d in range(2, watch_days+1):
HLC = get_hl_count_df(hl_list, date, d)
CHLC = HLC[HLC['count'] == d]
df = df.append(CHLC)
stock_list = list(set(df.index))
ccd = pd.DataFrame()
for s in stock_list:
tmp = df.loc[[s]] # 此处为什么要用双层方括号[[ ]],如果只用单层[ ],就会报错!
if len(tmp) > 1:
M = tmp['count'].max()
tmp = tmp[tmp['count'] == M]
ccd = ccd.append(tmp)
if len(ccd) != 0:
ccd = ccd.sort_values(by='count', ascending=False)
return ccd
2024-07-25