Update binary_count_trailing_zeros.py

The program prompts the user to enter a binary number.
It checks if the input contains only 0s and 1s using a generator expression.
If the input is valid, it calls the count_zeros_and_ones function and prints the results. If not, it notifies the user of the invalid input.
This commit is contained in:
USP-2024 2024-10-13 19:43:04 +05:30 committed by GitHub
parent e9e7c96465
commit f992b5eac6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -42,3 +42,25 @@ if __name__ == "__main__":
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
# Get user input
binary_number = input("Enter a binary number: ")
# Validate input
if all(bit in '01' for bit in binary_number):
zeros, ones = count_zeros_and_ones(binary_number)
print(f"Number of 0s: {zeros}, Number of 1s: {ones}")
else:
print("Invalid input! Please enter a valid binary number.")