使用pandas的DataFrame管理股池

股票数据推荐使用tushare获取。

推荐：https://tushare.pro/register?reg=133232 分享此链接，成功注册一个用户可获得50积分。

import pandas as pd

1、设置一个空的列表

security=[]

2、策略选股：使用pandas的DataFrame格式数据

以前是将选股结果打印出来,print.

现在将结果追加到空的列表

security.append([df.index[-1].date(),stock,stock_name,H,fall,HSL1.values()[-1],MAHSL2.values()[-1],MAHSL1.values()[-1]])
#print stock,stock_name

可以将感兴趣的信息都放到列表中，这里包括了股票被选中的日期df.index[-1].date(),股票代码、名称等又组成了一个列表。

3、将嵌套列表的列表转换成df

df_stock=pd.DataFrame(security)
df_stock.columns=list('abcdefgh')
df_stock.rename(columns={'a':'datetime', 'b':'stock','c':'stock_name',
                         'd':'下跌天数', 'e':'下跌区间跌幅','f':'换手','g':'五日平均换手',
                        'h':'下跌区间平均换手'}, inplace = True)

4、将DataFrame存储为csv,index表示是否显示行名，default=True

df_stock.to_csv("test.csv",index=False,sep=',')