From 7d38983528a2706efd404379126df203c9e13038 Mon Sep 17 00:00:00 2001 From: Administrator <admin@example.com> Date: 星期三, 28 五月 2025 14:57:44 +0800 Subject: [PATCH] 大单解析按照代码来聚合 --- data_parser/transaction_big_order_parser.py | 25 +++++++++++++------------ 1 files changed, 13 insertions(+), 12 deletions(-) diff --git a/data_parser/transaction_big_order_parser.py b/data_parser/transaction_big_order_parser.py index 31f3eb9..405432c 100644 --- a/data_parser/transaction_big_order_parser.py +++ b/data_parser/transaction_big_order_parser.py @@ -194,6 +194,9 @@ )) +__combined_df_cache = {} + + def extract_big_order_of_code(dir_path, code=None): """ 鎻愬彇浠g爜鐨勫ぇ鍗� @@ -224,15 +227,23 @@ print("鎷兼帴鏁版嵁涓嶅瓨鍦�") return + if not code: + codes = extract_big_order_codes(dir_path) + print("鎬讳唬鐮佹暟閲忥細", len(codes)) + for code in codes: + extract_big_order_of_code(dir_path, code) if code: output_path = os.path.join(dir_path, f"big_buy_{code}.csv") else: output_path = os.path.join(dir_path, f"big_buy.csv") if os.path.exists(output_path): - print("璺緞宸插瓨鍦�") + print("璺緞宸插瓨鍦�:", output_path) return - df = pd.read_csv(combined_path) + df = __combined_df_cache.get(combined_path) + if not df: + df = pd.read_csv(combined_path) + __combined_df_cache[combined_path] = df df_copy = df.copy() if code: df_copy = df_copy[df_copy["SecurityID"] == int(code)] @@ -257,16 +268,6 @@ @param code: @return: """ - - def first_last(group): - """ - 鑾峰彇绗竴鏉℃暟鎹笌鏈�鍚庝竴鏉� - @param group: - @return: - """ - return pd.Series({ - }) - combined_path = os.path.join(dir_path, 'combined.csv') if not os.path.exists(combined_path): print("鎷兼帴鏁版嵁涓嶅瓨鍦�") -- Gitblit v1.8.0