有没有一种简单的方法可以将一个或两个np.arrays传递给一个函数，而不会散布一个数组？

import numpy as np

def my_function(*args):
    if len(args) == 1:
        array1 = args[0]
        # 处理array1
        print("处理array1:", array1)
    elif len(args) == 2:
        array1 = args[0]
        array2 = args[1]
        # 处理array1和array2
        print("处理array1:", array1)
        print("处理array2:", array2)
    else:
        print("参数数量错误")

# 示例调用
array1 = np.array([1, 2, 3])
array2 = np.array([4, 5, 6])
my_function(array1)
my_function(array1, array2)

处理array1: [1 2 3]
处理array1: [1 2 3]
处理array2: [4 5 6]