a problem you could solve, or a situation you could simulate usuing classes is a library system.

dice_stats.py

@@ -19,6 +19,31 @@ class FiveDice:
                 return False
         return True
 
+    def is_three_of_a_kind(self):
+        face_counts = {}
+        for face in self.faces():
+            if face in face_counts:
+                face_counts[face] += 1
+            else:
+                face_counts[face] = 1
+        for count in face_counts.values():
+            if count >= 3:
+                return True
+        return False
+    
+    def is_four_of_a_kind(self):
+        face_counts = {}
+        for face in self.faces():
+            if face in face_counts:
+                face_counts[face] += 1
+            else:
+                face_counts[face] = 1
+        for count in face_counts.values():
+            if count >= 4:
+                return True
+        return False
+
+
 dice = FiveDice()
 successes = 0
 trials = 1000000
@@ -26,8 +51,20 @@ for trial in tqdm(range(trials)):
     dice.roll()
     if dice.all_ones():
         successes += 1
-
-print(successes/trials)
     
 
+print(f"Probability of all ones: {successes/trials}")
 
+three_of_a_kind_successes = 0
+for trial in tqdm(range(trials)):
+    dice.roll()
+    if dice.is_three_of_a_kind():
+        three_of_a_kind_successes += 1
+print(f"Probability of three of a kind: {three_of_a_kind_successes/trials}")
+
+four_of_a_kind_successes = 0
+for trial in tqdm(range(trials)):
+    dice.roll()
+    if dice.is_four_of_a_kind():
+        four_of_a_kind_successes += 1
+print(f"Probability of four of a kind: {four_of_a_kind_successes/trials}")