a
    cOjf                     @   sP   d Z ddlZddlmZmZ edddZejddd	Z	e
eed
ddZdS )u   
行业相关工具。
    N   )FINANCE_KEYWORDSINDUSTRY_CSV)returnc                  C   s4   t jtdtid} tt| d | d dtS )u6   加载行业分类，返回 {code: industry} 字典。codeZdtypeindustry )pdread_csvr   strdictzipfillnaZastype)df r   +/Users/admin/quant_ocf/ocf/core/industry.pyload_industry
   s    r   c                  C   sF   t jtdtid} dt}| d djj|dd| d< | g d	 S )
u?   加载行业分类 DataFrame，并添加 is_financial 标记。r   r   |r   r	   F)Znais_financial)r   r   r   )r
   r   r   r   joinr   r   contains)r   patr   r   r   load_industry_df   s    
r   )r   industry_mapr   c                    s"   | | d t fddtD S )u!   判断股票是否为金融类。r	   c                 3   s   | ]}| v V  qd S )Nr   ).0kr   r   r   	<genexpr>       zis_financial.<locals>.<genexpr>)getanyr   )r   r   r   r   r   r      s    r   )__doc__Zpandasr
   settingsr   r   r   r   Z	DataFramer   r   boolr   r   r   r   r   <module>   s
   