请用python写出求4个数最大值的程序代码。

下面是求4个数最大值的程序代码：

def find_max(num1, num2, num3, num4):
    max_num = num1
    if num2 > max_num:
        max_num = num2
    if num3 > max_num:
        max_num = num3
    if num4 > max_num:
        max_num = num4
    return max_num

print(find_max(1, 2, 3, 4))

该代码使用了一个名为find_max的函数，该函数接收四