Python:神奇的pivot函数!(行列转置)

是的,绵阳老板说得有道理。

做交易,绝大多数时间要盯着布朗运动看……

还是做实业,能静下心来,不管是做研究也好,或者做开发也好。

内心充实,不浮躁。


做组合回测,而数据又是从关系型数据库中来的,像下图这样。

Python:神奇的pivot函数!(行列转置)

但其实我更希望它摆成这样,才有助于向量化操作。

日期 601318.SH 600050.SH 600000.SH
2017/9/1 每一天的收盘价
2017/9/2
2017/9/3
2017/9/4
2017/9/5
2017/9/6
2017/9/7
2017/9/8
2017/9/9
2017/9/10


python有个pivot函数。


component_stk_md_df = pd.DataFrame(list(rs),columns=['stk_code','d_day','close'])

stk_close_df = component_stk_md_df.pivot(index='d_day', values='close', columns='stk_code')

# 索引 将是 日期, 列名 将是 代码, df中每一个单元格的值 将是 收盘价!

神奇…


好像mysql本身也可以行列转置,但是有点复杂,这一篇就算了,不写了。