CSZC_ZhilianBI/data_processing.py at main · persist-1/CSZC_ZhilianBI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
# data_processing.py
import re

def clean_salary(salary_str):
    """清洗薪资数据，转换为平均薪资（单位：千元）"""
    if not isinstance(salary_str, str) or not salary_str.strip():
        return None

    try:
        # 处理薪资范围
        if '-' in salary_str:
            if '万/月' in salary_str:
                parts = re.findall(r'(\d+\.?\d*)', salary_str)
                if len(parts) >= 2:
                    return (float(parts[0]) + float(parts[1])) / 2 * 10
            elif '千/月' in salary_str:
                parts = re.findall(r'(\d+\.?\d*)', salary_str)
                if len(parts) >= 2:
                    return (float(parts[0]) + float(parts[1])) / 2
            elif '万/年' in salary_str:
                parts = re.findall(r'(\d+\.?\d*)', salary_str)
                if len(parts) >= 2:
                    return (float(parts[0]) + float(parts[1])) / 2 / 12 * 10
            elif '元/天' in salary_str:
                parts = re.findall(r'(\d+\.?\d*)', salary_str)
                if len(parts) >= 2:
                    return (float(parts[0]) + float(parts[1])) / 2 * 0.022  # 按22天工作日换算
        # 处理面议或其它情况
        elif '面议' in salary_str or '议' in salary_str:
            return None
        # 处理固定薪资
        else:
            if '万/月' in salary_str:
                num = re.search(r'(\d+\.?\d*)', salary_str)
                if num:
                    return float(num.group(1)) * 10
            elif '千/月' in salary_str:
                num = re.search(r'(\d+\.?\d*)', salary_str)
                if num:
                    return float(num.group(1))
            elif '万/年' in salary_str:
                num = re.search(r'(\d+\.?\d*)', salary_str)
                if num:
                    return float(num.group(1)) / 12 * 10
            elif '元/天' in salary_str:
                num = re.search(r'(\d+\.?\d*)', salary_str)
                if num:
                    return float(num.group(1)) * 0.022  # 按22天工作日换算
        return None
    except Exception as e:
        print(f"薪资解析错误: {salary_str}, 错误: {e}")
        return None