如何在我的2D数组中显示最大数量？

def display_max_count(arr):
    max_val = float('-inf')  # 初始化最大值为负无穷大
    count = 0  # 初始化最大值的数量为0

    # 遍历整个2D数组，找到最大值和最大值的数量
    for row in arr:
        for val in row:
            if val > max_val:
                max_val = val
                count = 1
            elif val == max_val:
                count += 1

    # 创建新的2D数组，将最大值的数量显示在相应的位置上
    display_arr = [[count if val == max_val else val for val in row] for row in arr]

    return display_arr