Working on how to more efficiently apply density array

Data_Generation/Piecewise_Box_Functions.py

@@ -1,3 +1,4 @@
+import matplotlib.pyplot as plt
 import numpy as np
 from scipy import signal
 
@@ -53,3 +54,24 @@ def add_thickness(array_original, thickness):
         convolution = signal.convolve2d(A, filter, mode='same')
         A = np.where(convolution <= 1, convolution, 1)
     return A
+
+
+basic = basic_box_array(10, 1)
+densitys = [0.25,0,75,1]
+# multiple = np.repeat(basic, 3, axis=1 +)#np.multiply(basic,densitys)
+
+multiple = np.array([basic]*4, order='C')
+# multiple = np.concatenate(4*[[basic]], axis=0)
+# multiple = np.swapaxes(multiple, 0,2)
+# multiple = np.stack(4*[[basic]], axis=-1)
+
+# multiple = np.moveaxis(multiple, [0], [2])
+print(multiple)
+print(np.shape(multiple))
+multiple = np.multiply(densitys, multiple)
+print(np.shape(multiple))
+# print(multiple)
+plt.imshow((multiple[0]))
+plt.show()
+
+