2022-02-14 17:28:52 +08:00
|
|
|
import pandas as pd
|
|
|
|
from company.company_db import FIND_INDUSTRY_DATA_IN_RATING_INPUT
|
|
|
|
|
|
|
|
|
2022-01-27 17:01:38 +08:00
|
|
|
def sub_dict(param1, param2):
|
|
|
|
"""
|
|
|
|
获取字典的子集
|
|
|
|
Parameters:
|
|
|
|
param1: 原字典
|
|
|
|
param2: 子集字段
|
|
|
|
Returns:
|
|
|
|
子集
|
|
|
|
"""
|
2022-02-14 17:28:52 +08:00
|
|
|
return dict((key, value) for key, value in param1.items() if key in param2)
|
|
|
|
|
|
|
|
|
|
|
|
def judge_industry(cid):
|
|
|
|
"""
|
|
|
|
判断公司所属行业
|
|
|
|
"""
|
|
|
|
|
|
|
|
def judge_by_input_form():
|
|
|
|
input_form_data = FIND_INDUSTRY_DATA_IN_RATING_INPUT(cid)
|
|
|
|
|
|
|
|
if not input_form_data:
|
|
|
|
return False
|
|
|
|
|
|
|
|
industry = pd.DataFrame(input_form_data).sort_values('评价年度', ascending=False)[0:1]['行业选择'].values[0][0]
|
|
|
|
return industry
|
|
|
|
|
|
|
|
return judge_by_input_form()
|