Counting 0s and 1s:Update binary_count_trailing_zeros.py

bit_manipulation/binary_count_trailing_zeros.py

@@ -42,3 +42,20 @@ def binary_count_trailing_zeros(a: int) -> int:
     import doctest
 
     doctest.testmod()
+
+
+# counting number of 0s and 1s in a binary number
+def count_zeros_and_ones(binary_number):
+    # Convert the binary number to a string if it's not already
+    binary_str = str(binary_number)
+
+    count_zeros = binary_str.count("0")
+    count_ones = binary_str.count("1")
+
+    return count_zeros, count_ones
+
+
+# Example usage
+binary_number = 1011001
+zeros, ones = count_zeros_and_ones(binary_number)
+print(f"Number of 0s: {zeros}, Number of 1s: {ones}")