如何覆盖数组的打印？

def print_array(arr):
    for element in arr:
        print(element)

array = [1, 2, 3, 4, 5]
print_array(array)

1
2
3
4
5

def print_array(arr):
    print(' '.join(map(str, arr)))

array = [1, 2, 3, 4, 5]
print_array(array)

1 2 3 4 5