From 411e1a3f783db7c5b9e21ed03df1653f5b8b1782 Mon Sep 17 00:00:00 2001
From: binz <123@123.com>
Date: Sun, 16 Jun 2024 23:31:39 +0800
Subject: [PATCH] =?UTF-8?q?Update:=20=E6=96=B0=E5=A2=9E=E6=8C=87=E5=AE=9A?=
 =?UTF-8?q?=E9=87=8F=E5=88=86=E9=92=9F=E4=BB=B7=E6=A0=BC=E8=AE=A1=E7=AE=97?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Specified_Price.py | 33 +++++++++++++++++++++++++++++++++
 __init__.py        |  3 ++-
 2 files changed, 35 insertions(+), 1 deletion(-)
 create mode 100644 Specified_Price.py

diff --git a/Specified_Price.py b/Specified_Price.py
new file mode 100644
index 0000000..2cc3d73
--- /dev/null
+++ b/Specified_Price.py
@@ -0,0 +1,33 @@
+import pandas as pd
+import numpy as np
+
+
+def amount_specified(
+    min_data: pd.DataFrame,
+    post_adj_factor: pd.DataFrame,
+    min_amount: float=0.
+):
+    """
+    计算指定金额下的平均后复权价格
+    
+    Args:
+    min_data (pd.DataFrame):        指定日期分钟数据，需至少包含代码(stock_code)、分钟(Time)、价格(price)、成交量(vol)、成交金额(amount)列
+    post_adj_factor (pd.DataFrame): 后复权因子数据，需至少包含股票、后复权因子
+    min_amount (float):             指定最小金额下的平均价
+    """
+    
+    # 按照指定最小量获取平均价格
+    stock_amt = min_data.pivot_table(index='Time', columns='stock_code', values='amount').cumsum()
+    stock_vol = min_data.pivot_table(index='Time', columns='stock_code', values='vol').cumsum()
+    amount_price = stock_amt / stock_vol
+    amount_price.iloc[1:] = amount_price.iloc[1:].where(stock_amt <= min_amount, np.nan)
+    amount_price = amount_price.unstack().reset_index()
+    amount_price.columns = ['stock_code', 'time', 'price']
+    amount_price = amount_price.dropna(subset=['price']).drop_duplicates(subset='stock_code', keep='last')
+    # 计算后复权价格
+    amount_price = amount_price.merge(post_adj_factor, on=['stock_code'], how='left').dropna(subset=['factor'])
+    amount_price['open_post'] = amount_price['price'] * amount_price['factor']
+
+    return amount_price[['stock_code','open_post']]
+    
+    
\ No newline at end of file
diff --git a/__init__.py b/__init__.py
index 45a5f56..3a33c7c 100644
--- a/__init__.py
+++ b/__init__.py
@@ -2,4 +2,5 @@ from account import Account
 from trader import Trader
 from spread_backtest import Spread_Backtest
 
-__all__ = ['Account', 'Trader', 'Spread_Backtest']
\ No newline at end of file
+
+__all__ = ['Account', 'Trader', 'Spread_Backtest', 'Specified_Price']
\ No newline at end of file