自己做量化交易软件(15)通通量化AI框架的选股器设计
前面我们学会了单只股票的分析，回测功能。下面我们开始设计选股器和量化分析的功能。
在设计选股器之前，我们先要学习一些python基本操作技巧。
一、股票代码处理技巧
1、数字类型转换
我们获取的数据格式都不相同，特别是股票代码，有些是数字1，有些是字符1，有些股票代码带后缀，例如’60000.SH’，‘000001.SZ’，‘000001.XSHE’，'000300.XSHG’等等。无论什么方式，我们都做一些简单的转化为通通标准股票代码格式。
深圳股票代码： ‘000001.SZ’
上海股票代码： ‘600000.SH’
指数代码：‘000300.ZS’
各种数据转化函数命令如下：

#数字int转换为字串str
s_code=str(n_code) #字串str转换为数字int
n_code=int(s_code)

代码演示：

n_code=10
#数字int转换为字串str
s_code=str(n_code) 
print(s_code,type(s_code))
#字串str转换为数字int
n_code=int(s_code)
print(n_code,type(n_code))

输出结果：
10 <class ‘str’>
10 <class ‘int’>

2、字符串处理
我们看到字串’10’ ，不够长度6，需要在前面补’0’。
字串数值前面补0，用函数zfill()，看下面演示代码和结果：

s_code='10'
#字串数字前补够0
s_code= s_code.zfill(6)

输出结果:
000010 <class ‘str’>

3、补上股票后缀
看下面股票代码演示：

#通通股票代码转换
def ttsn(s):s=s.strip()if (len(s)<6 and len(s)>0):s=s.zfill(6)+'.SZ'if len(s)==6:if s[0:1]=='0':s=s+'.SZ'else:s=s+'.SH'return s#聚宽股票代码转换
def jqsn(s):s=s.strip()if (len(s)<6 and len(s)>0):s=s.zfill(6)+'.XSHE'if len(s)==6:if s[0:1]=='0':s=s+'.XSHE'else:s=s+'.XSHG'return s

二、股票数据中的股票代码处理技巧
pandas是基于numpy构建的，为时间序列分析提供了很好的支持。pandas中有两个主要的数据结构，一个是Series，另一个是DataFrame。
Series 类似于一维数组与字典(map)数据结构的结合。它由一组数据和一组与数据相对应的数据标签（索引index）组成。这组数据和索引标签的基础都是一个一维ndarray数组。可将index索引理解为行索引。 Series的表现形式为：索引在左，数据在右。
DataFrame是一个类似表格的数据结构，索引包括列索引和行索引，包含有一组有序的列，每列可以是不同的值类型（数值、字符串、布尔值等）。DataFrame的每一行和每一列都是一个Series，这个Series的name属性为当前的行索引名/列索引名。
股票数据一般用pandas包的 DataFrame 来存放。
上述的方法无法对 DataFrame整列数据处理。用循环操作又充分发挥了Python的速度慢的弱点，因此可以用一些Python的技巧来提高速度。
DataFrame序列中的股票格式转换可以用类似下面的命令来瞬间完成。

#短字串补全快速运算
base.code=['0'*(6-len(x)) + x for x in base.code.astype(str)]

三、股票代码的运算
我们通过若干算法，例如财务数据或者技术指标等策略， 筛选了一批数据。怎样通过这些数据建立我们的目标备选股票池。并且完成股票池之间的逻辑运算。
例如 股票池的合并、黑名单的过滤等等运算。我们可以充分利用Python的特征和优势，来快速完成股票池运算。
我们可以充分利用这些不同数据特性，快速完成所需工作。
1、pandas和numpy数据相互转换
下面给出DataFrame中的股票代码转列表例子，重点要看说明和记住例子，演示代码如下：

import pandas as pd
import numpy as np
import HP_data as hp
#获取业绩报告数据
'''
report_data返回数据格式 如下：
code,代码
name,名称
esp,每股收益
eps_yoy,每股收益同比(%)
bvps,每股净资产
roe,净资产收益率(%)
epcf,每股现金流量(元)
net_profits,净利润(万元)
profits_yoy,净利润同比(%)
distrib,分配方案
report_date,发布日期
'''#获取获取2015年第4季度的业绩报告数据
t2015=hp.get_report_data(2015,4) #选取eps_yoy,每股收益同比(%)的排名前20位股票数据。
top20=t2015.sort_values(by='eps_yoy',ascending=False).head(20)#获取股票代码的Series序列，带原始数据索引
S_code=top20.code#dataframe重建索引，从0开始顺序
S_code2 = S_code.reset_index(drop=True)#将股票代码Series序列转为列表list数据
l_code=S_code.tolist()#将列表list转为array序列数据，与Series序列区别是无索引。
a_code=np.array( l_code)#将array序列数据转为 Series序列，增加了从0开始顺序自然索引。
S_code3= pd.Series(a_code, name = 'code')#部分数据内容输出结果
print('\n----S_code-----获取股票代码的Series序列，带原始数据索引')
print(S_code)
print('\n----S_code2-----重建索引，从0开始顺序')
print(S_code2)
print('\n----l_code-----将股票代码Series序列转为列表list数据')
print(l_code)
print('\n----a_code-----将列表list转为array序列数据，与Series序列区别是无索引。')
print(a_code)
print('\n----S_code3-----将array序列数据转为 Series序列，增加了从0开始顺序自然索引。')
print(S_code3)

程序运行结果如下图：

2、列表（list）和集合（set）相互转换
Python数据格式有列表（list）和集合（set）。
列表有重复数据，并且 有序。
集合无重复数据，并且无序。
我们可以利用集合(set)的无重复特性，将股票代码转化为集合(set)，能够快速完成股票代码去掉重复，股票代码池的合并，剔除股票黑名单等运算。
（1） 列表运算

#假定我们通过策略1筛选出股票池列表bk1
bk1=['600000', '600001', '600002', '600003', '600004', '000001', '000010']#我们通过策略2筛选出股票池列表bk2，我们可以看到前面3个数据重复。
bk2=['600000', '600001', '600002','600981',  '600080', '600191']#给bk1查入数据数据,list.insert(i, 数据) 
bk1=bk1.insert(1, '300751')    #在指定位置插入元素#给bk2增加数据
bk2=bk2.append('600390')#删除bk2最后一个数据
bk2=bk2.pop()    #将最后一位的元素删除#删除bk1中间第3个数据。
bk1=bk1.pop(3)    #删除指定位置的元素

（2） 集合运算
python的set和其他语言类似, 是一个无序不重复元素集, 基本功能包括关系测试和消除重复元素. 集合对象还支持union(联合), intersection(交), difference(差)和sysmmetric difference(对称差集)等数学运算。
sets 支持 x in set, len(set),和 for x in set。作为一个无序的集合，sets不记录元素位置或者插入点。因此，sets不支持 indexing, slicing, 或其它类序列（sequence-like）的操作。
集合t,s支持一系列标准操作，包括并集、交集、差集和对称差集，例如：
a = t | s # t 和 s的并集
b = t & s # t 和 s的交集
c = t – s # 求差集（项在t中，但不在s中）
d = t ^ s # 对称差集（项在t或s中，但不会同时出现在二者中）
下面给出集合运算的演示。

#建立集合st1,st2
print('\n----st1----')
st1=set(['002027', '600061', '600080','600659'])
print(st1)print('\n----st2----')
st2=set(['000001', '600061', '600088','000002'])
print(st2)#集合中增加可以通过列表list增加，再转换为集合set。
#也可以通过集合并集运算来增加元素。
print('\n----st1---- st1中增加600090')
st1=st1.union(set(['600090']))
print(st1)print('\n----a = st1 | st2-----')
#集合运算
a = st1 | st2   # st1 和 st2的并集
print(a)print('\n----b = st1 & st2 -----')
b = st1 & st2  # st1和 st2的交集
print(b)print('\n----c = st1 - st2 -----')
c = st1 - st2   # 求差集（项在st1中，但不在st2中）
print(c)print('\n----d = st1 ^ st2-----')
d = st1 ^ st2  # 对称差集（项在st1或st2中，但不会同时出现在二者中）
print(d)print('\n----st2 remove 600088-----')
#集合元素删除
st2=st2.difference(set(['600088']))
print(st2)
print(type(st1),type(st2))

程序运行结果如下图：

（3）列表和集合转换
我们根据算法需要，利用列表和集合的特性，来实现我们的要求

#假定我们通过策略1筛选出股票池列表bk1，但又股票重复
bk1=['600000', '600001', '600002', '600001', '600000', '000001', '000010','000002']
print('\n-----列表bk1----')
print(bk1)print('\n-----集合A----')
#转换为集合A来去掉重复
A=set(bk1)
print(A)print('\n-----列表bk2----')
#将结果转为列表bk2
bk2=list(A)
print(bk2)print('\n-----列表bk2排序----')
#对bk2进行排序
bk3=bk2.sort()
print(bk2)

程序运行结果如下图：

今天我给大家介绍了关于股票代码和股票板块运算的基本知识。下次开始设计有关选股器画面的设计。