a
    l#Ojf                     @   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dtypeindustry )pdread_csvr   strdictzipfillnaastype)df r   0/tmp/quant_ocf_run_20260709/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)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__pandasr   settingsr   r   r   r   	DataFramer   r   boolr   r   r   r   r   <module>   s
   