From c250180ef73ec16963790f1c814e0db365409f80 Mon Sep 17 00:00:00 2001 From: Tiger Ren Date: Wed, 21 Aug 2024 17:42:32 +0800 Subject: [PATCH] added error handling --- analysis_action_result.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/analysis_action_result.py b/analysis_action_result.py index fec454a..3b829a1 100644 --- a/analysis_action_result.py +++ b/analysis_action_result.py @@ -1,7 +1,10 @@ import re import pandas as pd +import math from AgentProxy import AgentProxy +def is_none_or_nan(variable): + return variable is None or (isinstance(variable, (int, float)) and math.isnan(variable)) result_category_list = [ "完成需求确认与收集", @@ -22,7 +25,10 @@ data = [] for index, row in df.iterrows(): text = row["行动结果"] - # print(text) + if is_none_or_nan(text): + print(f"index{index} ") + continue + action_pattern = re.compile(r'\*\*销售动作\*\*:(.+?)\n') category_pattern = re.compile(r'\*\*销售动作归类\*\*:(.+?)\n') result_pattern = re.compile(r'\*\*销售动作行动结果\*\*:(.+?)\n') @@ -33,9 +39,9 @@ for index, row in df.iterrows(): results = result_pattern.findall(text) abstract_results = abstract_result_pattern.findall(text) if (len(actions) == len(categories) and len(actions) == len(results) and len(actions) == len(abstract_results)): - print("index result compliance: true") + print(f"index{index} result compliance: true") else: - print("index result compliance: false") + print(f"index{index} result compliance: false") continue # Combine into a list of dictionaries