添加策略开始测试

2025-09-10 00:59:13 +08:00
parent a338f50867
commit e75aca320e
5 changed files with 48 additions and 33 deletions
@@ -81,6 +81,26 @@ class ChanLun():
 	def plot_dataframe(self, dataframe):
 		klc_list = self.get_klc_list(dataframe)
 		bi_list= self.cal_bi_list(klc_list)
+	def get_klu_state_list(self, dataframe):
+		klu_list = self.get_klu_list(dataframe)
+		chanmacd = ChanMACD(klu_list)
+		state_list = []
+		for klu in klu_list:
+			if klu.macd > 0:
+				if klu.separate_div:
+					state_list.append("20")
+				elif klu.continue_div:
+					state_list.append("20")
+				else:
+					state_list.append("00")
+			else:
+				if klu.separate_div:
+					state_list.append("-20")
+				elif klu.continue_div:
+					state_list.append("-20")
+				else:
+					state_list.append("00")
+		return state_list
 	def get_klc_state_list(self, dataframe):
 		klc_list = self.get_klc_list(dataframe)
 		bi_list= self.cal_bi_list(klc_list)