Python分析三季度基金调仓
作者：互联网
目前三季度已经过去了一个月，大部分基金都已经公布了三季度持仓数据，今天我们就用 Python 分析以下今年三个季度基金的调仓情况。
获取数据
第一步，我们要获取目前发行的所有基金及其持有的股票。
可以写一个爬虫去基金网站爬数据，但太麻烦，这里其实是有捷径的。有些朋友可能听说过量化投资，这些做量化投资的平台都会提供金融数据，我们只要安装相应的 Python 包就可以获取股票、基金、债券相关的数据。
本次分析用的是聚宽（https://www.joinquant.com/）平台，新用户注册后会有6个月试用期，期间可以免费使用平台所有数据，每天可调用100万条数据，完全够我们分析了。
注册账号后，我们就可以调用聚宽的数据。以下代码我用 Python 3.8+jupyter 编写、运行。
导入包，获取聚宽授权
from jqdatasdk import *
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
plt.rcParams['font.sans-serif'] = ['Arial Unicode MS'] # mac matplot显示中文
plt.rcParams['axes.unicode_minus'] = False
auth('聚宽账户', '聚宽密码') # 聚宽授权
调用聚宽 get_all_securities 函数，获取所有的基金，该函数文档如下
代码如下：
df = get_all_securities(['fund', 'open_fund'], '2021-11-10')
df
df 里可能会有重复基金，按照基金代码（df.index）去重看看共获取了多少只基金
code_arr = list(set([x.split('.')[0] for x in df.index.values]))
len(code_arr)
共返回 13698 支基金，这可比爬虫来得快多了。
获取股票投资占比
这里我只想关注股票类型的基金，所以需要获取每支基金股票投资的占比，股票投资占比小于 50% 的，剔除掉。
从文档来看，查询 FUND_PORTFOLIO 表的 stock_rate 字段就可以获取基金的股票投资占比。查询方式是用 query 函数构造查询语句，然后调用 finance.run_query 函数完成查询。
写一个函数，用来返回查询语句
def asset_query(arr):
q=query(finance.FUND_PORTFOLIO.code,
finance.FUND_PORTFOLIO.name,
finance.FUND_PORTFOLIO.period_end,
finance.FUND_PORTFOLIO.report_type,
finance.FUND_PORTFOLIO.stock_rate
).filter(finance.FUND_PORTFOLIO.code.in_(arr),
finance.FUND_PORTFOLIO.period_end.in_(['2021-03-31', '2021-06-30', '2021-09-30']),
finance.FUND_PORTFOLIO.report_type.in_(['第一季度', '第二季度', '第三季度']))
return q
asset_query 函数的参数 arr 是个数组，存放基金代码。query函数指定返回的字段。filter 函数指定需要筛选的数据，code 代表基金代码；period_end 是报告期（每个季度最后一天），填入的三个日期分别代表每个季度；report_type 是报告类型，填入的是三个季度，添加该条件是为了只读取单季度的数据，不需要半年或年度数据。最终返回查询语句 q。
asset_query 函数生成对象 q 的用的是 SQLAlchemy 提供的语法规则，SQLAlchemy 是Python 中著名的 ORM 框架，简单来说它提供以 Python 代码的方式查询数据库，而不用写 SQL。感兴趣的朋友可以学习下，能够帮你实现更复杂的查询逻辑。
运行 finance.run_query 来执行查询
i = 0
while i < len(code_arr):
print('获取基金资产 ' + str(i))
tmp_arr = code_arr[i: i+1500]
q = asset_query(tmp_arr)
tmp_df = finance.run_query(q)
if i == 0:
asset_df = tmp_df
else:
asset_df = pd.concat([asset_df, tmp_df])
i += 1500
因为 finance.run_query 单次最多返回 5000 行数据，所以这里写了个循环分批读取，一次只读取 1500 个基金的数据。因为一个基金返回 3 个季度数据，所以一次读取最多返回 4500 数据，不会超限。
读取完成后，看看 asset_df
筛选 stock_rate 大于 50 的基金
stock_fund_df = asset_df[asset_df['stock_rate'] > 50]
统计筛选后，还剩多少只基金
stock_fund_code_arr = list(set(stock_fund_df['code']))
len(stock_fund_code_arr)
返回 5590，过滤掉了一半。
获取基金持仓
获取基金持仓的方式跟上面一样，只不过表名和字段名变了。
编写 hold_stock_query 函数，生成相应的查询语句
def hold_stock_query(arr):
q=query(finance.FUND_PORTFOLIO_STOCK
).filter(finance.FUND_PORTFOLIO_STOCK.code.in_(arr),
finance.FUND_PORTFOLIO_STOCK.rank <= 10,
finance.FUND_PORTFOLIO_STOCK.period_end.in_(['2021-03-31', '2021-06-30', '2021-09-30']),
finance.FUND_PORTFOLIO_STOCK.report_type.in_(['第一季度', '第二季度', '第三季度']))
return q
finance.FUND_PORTFOLIO_STOCK.rank <= 10代表只获取前十大重仓股。
同样通过循环的方式查询数据
i = 0
while i < len(stock_fund_code_arr):
print('获取基金持仓 ' + str(i))
tmp_arr = stock_fund_code_arr[i: i+150]
q = hold_stock_query(tmp_arr)
tmp_df=finance.run_query(q)
if i == 0:
hold_stock_df = tmp_df
else:
hold_stock_df = pd.concat([hold_stock_df, tmp_df])
i += 150
hold_stock_df
看看 hold_stock_df 的信息
name是持有的股票名，rank是持有该股票的排名（1 代表该基金第一大重仓股），proportion是持仓该股票比例。
调仓方向
这里我想按照行业来看，如：从第一季度到第三季度，有多少基金购买了新能源股票，有多少基金购买了白酒股票。所以，就需要将股票对应到行业上。
严格来说应该找权威的券商，将每个股票都映射成某个行业。不过这里我只是想定性分析一下，所以找了一些行业指数基金的前 6 大重仓股来代替某个行业。映射关系如下
stock_to_industry_dict = {
'比亚迪':'新能源',
'宁德时代':'新能源',
'恩捷股份':'新能源',
'赣锋锂业':'新能源',
'亿纬锂能':'新能源',
'汇川技术':'新能源',
'圣邦股份':'半导体',
'韦尔股份':'半导体',
'卓胜微':'半导体',
'北方华创':'半导体',
'兆易创新':'半导体',
'三安光电':'半导体',
'贵州茅台':'白酒',
'五粮液':'白酒',
'山西汾酒':'白酒',
'泸州老窖':'白酒',
'洋河股份':'白酒',
'酒鬼酒':'白酒',
'阳关电源':'光伏',
'通威股份':'光伏',
'中环股份':'光伏',
'隆基股份':'光伏',
'特变电工':'光伏',
'正泰电器':'光伏',
'药明康德':'医疗',
'智飞生物':'医疗',
'沃森生物':'医疗',
'泰格医药':'医疗',
'长春高新':'医疗',
'凯莱英':'医疗',
'复星医药':'医疗',
}
向 hold_stock_df 添加一列 industry，代表持有该股票所对应的行业
def stock_to_industry(cols):
if cols['name'] in stock_to_industry_dict:
return stock_to_industry_dict[cols['name']]
return '无'
hold_stock_df['industry'] = hold_stock_df.apply(lambda x: stock_to_industry(x), axis=1)
hold_stock_df
找一个新能源基金看下效果
hold_stock_df[hold_stock_df['code'] == '005939']
有了行业信息，就可以用透视表直接统计每个行业在三个季度基金数量的变化
industry_count_df = hold_stock_df.pivot_table(index='industry',
columns=['report_type'],
values='code',
aggfunc=lambda x:len(x.unique()))
cols_name = ['第一季度', '第二季度', '第三季度']
industry_count_df = industry_count_df[cols_name]
industry_count_df.drop(['无'], inplace=True)
plt.rcParams['figure.figsize'] = (12, 8)
industry_count_df.plot.bar()
从图上还是能发现一些明显的特征的，持仓新能源的基金不光基数大，增速速度还很快。比较意外的是白酒的持仓基金数量居然是下降的。
探索更有意思的
由于我们能够拿到所有的基金，其实我们还可以探索的东西有很多。比如：可以做出如下效果的数据
每一行都代表一只基金，列 rank1 ~ rank10 代表该基金前十大重仓股，逗号前半部分是持仓股票，后半部分是持仓占比。标红代表是我们关注的股票，最后一列是标红股票持仓占比的加和。
这种展示方式有很多好处，比支付宝、同花顺软件还好。
第一，支持按照某些股票筛选基金（上图标红的部分），我了解的基金软件都只能按照一只股票筛选基金
第二，可以计算包含这些股票总持仓占比（candid_prop列），总持仓越高跟我们预期越接近。而现有的基金软件都要自己手动加
第三，一行展示基金所有持仓，一目了然，而基金软件看不同基金的持仓需要在不同基金之间来回切页面，非常不方便。
最后，介绍下上图的实现代码。首先，只筛选最新季度持仓即可
hold_stock_p3_df = hold_stock_df[hold_stock_df['report_type'] == '第三季度']
构造两个新列
hold_stock_p3_df['hold_rank'] = ['rank%d' % i for i in hold_stock_p3_df['rank']]
hold_stock_p3_df['hold_info'] = hold_stock_p3_df['name'] + " : " + hold_stock_p3_df['proportion'].astype(str)
用透视图，列转行
tmp_df = hold_stock_p3_df.pivot(index='code', columns='hold_rank', values='hold_info')
tmp_idx = tmp_df.index
tmp_df = tmp_df.reset_index()
tmp_df['name'] = tmp_idx
rank_cols = ['rank%d' % i for i in range(1, 11)]
col_names = ['code'] + rank_cols
final_fund_df = tmp_df[col_names]
final_fund_df
定义函数，计算候选股票集合的总持仓占比
candid_fund_list = ['宁德时代','比亚迪', '恩捷股份', '璞泰来', '诺德股份']
def filter_fund(x):
all_prop = 0
for i in range(1, 11):
col = 'rank' + str(i)
vals = str(x[col]).split(':')
if vals[0].strip() in candid_fund_list:
all_prop += float(vals[1].strip())
return all_prop
定义函数用于上色
def show_color(val):
color = '#BB0000' if str(val).split(':')[0].strip() in candid_fund_list else ''
return 'color:%s' % color
修改 condid_fund_list 的值，就可以股票筛选相关的基金，实现本节开头那张图的效果
final_fund_df['candid_prop'] = final_fund_df.apply(lambda x: filter_fund(x), axis=1)
tmp_fund_df = final_fund_df[final_fund_df['candid_prop'] > 0].sort_values(by='candid_prop', ascending=False)
tmp_fund_df = tmp_fund_df.style.applymap(show_color)
tmp_fund_df
这里我们只分析了基金数据，聚宽平台还包括股票、债券等其他金融数据，获取方式跟上面讲的一样。有兴趣的朋友可以试试。
源文件已经整理好，关注公众号 渡码 回复关键词 基金调仓 即可。
标签：tmp,code,fund,Python,三季度,df,调仓,hold,stock 来源： https://www.cnblogs.com/duma/p/15566711.html