如何编写以Python字节数组为参数并计算比例的函数

def calculate_ratio(byte_array):
    total_bytes = len(byte_array)
    zero_count = byte_array.count(0)
    non_zero_count = total_bytes - zero_count
    
    zero_ratio = zero_count / total_bytes
    non_zero_ratio = non_zero_count / total_bytes
    
    return zero_ratio, non_zero_ratio

byte_array = [0, 1, 0, 0, 1, 1, 0, 1, 0, 1]
zero_ratio, non_zero_ratio = calculate_ratio(byte_array)
print("零字节比例：", zero_ratio)
print("非零字节比例：", non_zero_ratio)

零字节比例： 0.6
非零字节比例： 0.4