barplot function added

JosePabloGFnl · Aug 22, 2024 · 70b0c97 · 70b0c97
1 parent 387ec6a
commit 70b0c97
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/plotting.py b/plotting.py
@@ -13,6 +13,8 @@ def plot(self):
             return self.histogram()
         elif self.plot_option == 2:
             return self.boxplot()
+        elif self.plot_option == 4:
+            return self.barplot()
         else:
             return "Invalid plot option"
 
@@ -26,4 +28,10 @@ def boxplot(self):
         self.data.boxplot(column=self.x)
         plt.title("Boxplot of " + self.x)
         plt.show()
+        return self.data[self.x].describe()
+
+    def barplot(self):
+        pd.DataFrame(self.data[self.x].value_counts(normalize=True)).plot.barh()
+        plt.title("Bar plot of "+x)
+        plt.show()
         return self.data[self.x].describe()