update plot for custom_bollinger_rsi to stacked

tetrahedral-tech · Nov 18, 2023 · 3db4785 · 3db4785
1 parent aa5a0a1
commit 3db4785
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/plots/custom_bollinger_rsi.py b/plots/custom_bollinger_rsi.py
@@ -7,7 +7,7 @@
 import plots.colors as colors
 
 def plot(prices):
-	gs = GridSpec(2, 2, figure=plt.gcf())
+	gs = GridSpec(3, 1, figure=plt.gcf())
 
 	plt.subplot(gs[0, :])
 	indicies = np.arange(0, prices.shape[0])
@@ -22,9 +22,9 @@ def plot(prices):
 	plt.scatter(indicies[upper_condition], sliced_prices[upper_condition], color=colors.upper())
 	plt.scatter(indicies[lower_condition], sliced_prices[lower_condition], color=colors.lower())
 
-	plt.subplot(gs[-1, :1])
+	plt.subplot(gs[-1, :])
 	rsi(prices)
-	plt.subplot(gs[-1, 1:2])
+	plt.subplot(gs[-2, :])
 	bollinger_bands(prices)
 
 	return plt