请问有什么办法可以快速统计每期新入的股票数量?

Question

我的表结构：

trade_dt, security_id
2020.01.11   00001
2020.01.11   00002
2020.01.11   00003
2020.01.12   00002
2020.01.12   00003
2020.01.12   00004
···

请问有什么办法可以快速统计每期新入的股票数量？

Answer 1

可以尝试一下下面的代码：

def calcNewStockNum(dates, symbols){
 d = dates.groups()
 sortedDates = d.keys().sort()
 dailyStocks = loop(x->symbols[d[x]].set(), sortedDates)
 newStockNum = eachPre(def(t2, t1)->size(t2 - t1), dailyStocks).flatten()
 return table(sortedDates as date, newStockNum)
}

calcNewStockNum(t.date, t.symbol)