迅投xtquant获取股票对应的申万行业
官网中板块/行业对应的教程:https://dict.thinktrader.net/dictionary/industry.html?id=S7b19l
这里我们使用以下具体的函数:
xtdata.get_sector_list()
:这个代码获取全部的板块信息xtdata.get_stock_list_in_sector(板块名称)
:这个代码获取板块对应的股票列表
整理后的方法如下:
from xtquant import xtdata
import pandas as pddef get_industry_by_stock() -> pd.DataFrame:"""获取股票对应的行业"""reverse_industry = {}sw_sectors = [sector for sector in xtdata.get_sector_list() if (sector.startswith('SW1') and ("加权" not in sector))]for _sector in sw_sectors:for _stk in xtdata.get_stock_list_in_sector(_sector):reverse_industry[_stk] = _sectorreturn pd.get_dummies(pd.Series(reverse_industry))industry_df = get_industry_by_stock()
print(industry_df.head())
可以得到如下结果: