Python实现窗口函数

计算每只股票持仓前五的用户信息

SQL实现

select * from 
(
select sec_code,acct_id,hold_vol,hold_amt,DENSE_RANK() OVER(partition by sec_code order by hold_amt desc) as rank 
from sec_hold 
) a
where a.rank <=5

输出
在这里插入图片描述

#Python实现

def query_sql(sql):
    conn = psycopg2.connect(database='quant', host='10.211.55.20', port='5432', user='kiwi',password='yx3239736')
    conn.set_client_encoding('utf-8')
    cursor = conn.cursor()
    cursor.execute(sql)
    rows = cursor.fetchall()
    df = pd.DataFrame(rows)
    return df

def group_func(dataFame):
    return dataFame.sort_values(['hold_amt'],ascending=False)[:5]


sql = "select sec_code,acct_id,hold_vol,hold_amt from sec_hold "
df = query_sql(sql)
df.rename(columns={0:'sec_code', 1:'acct_id', 2:'hold_vol',3:'hold_amt',4:'rank'}, inplace = True)

df1 = df.groupby('sec_code').apply(group_func)

print(df1.head(30))

输出
在这里插入图片描述
结果一致

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值