Found error in reading of output files. Need to edit how to output the files

Data_Generation/Dataset_Generation_Functions.py

@@ -1,5 +1,6 @@
 import numpy as np
-
+import pandas as pd
+import json
 # from Data_Generation.Shape_Generation_Functions import basic_box, diagonal_box_split, horizontal_vertical_box_split, \
 #     back_slash_box, forward_slash_box, back_slash_plus_box, forward_slash_plus_box, hot_dog_box, hamburger_box, \
 #     x_hamburger_box, x_hot_dog_box, x_plus_box
@@ -70,4 +71,34 @@ def make_boxes(image_size, densities):
 
 
 ########################################################################################################################
+image_size = 9
+densities = [0.2, 0.4, 0.6, 0.8, 1]
+boxes = make_boxes(image_size, densities)
+
+box_arrays, box_density, basic_box_thickness, forward_slash_box_thickness, back_slash_box_thickness,hot_dog_box_thickness, hamburger_box_thickness\
+    = list(zip(*boxes))[0], list(zip(*boxes))[1], list(zip(*boxes))[2], list(zip(*boxes))[3], list(zip(*boxes))[4], list(zip(*boxes))[5], list(zip(*boxes))[6]
+
+
+
+# Create a dataframe to convert the data to a csv file
+dataframe = (pd.DataFrame((box_arrays, box_density, basic_box_thickness, forward_slash_box_thickness, back_slash_box_thickness,hot_dog_box_thickness, hamburger_box_thickness)).T).astype(str)
+
+# Rename the columns to the desired outputs
+dataframe = dataframe.rename(columns={0: "Array", 1: "Density", 2:"Basic Box Thickness", 3:"Forward Slash Strut Thickness", 4:"Back Slash Strut Thickness", 5:"Vertical Strut Thickness", 6:"Horizontal Strut Thickness"})
+
+csv = dataframe.to_csv('2D_Lattice')
+
+
+
+df = pd.read_csv('2D_Lattice')
+# print(df)
+row = 0
+box = df.iloc[row, 1]
+print(box)
+array = np.array(json.loads(box))
+
 
+# import matplotlib.pyplot as plt
+# plt.imshow(np.array(box))
+# plt.show()
+# print(df)