定义函数中参数为多维数据

Question

怎么在Python中定义函数中的参数是多维数据

def EPflux_quasigeostrophic(U,V,T,time_name=None, lat_name=None, lon_name=None, level_name=None):
    if lat_name is None:
        lat_name = utils.get_lat_name(U)
    if lon_name is None:
        lon_name = utils.get_lon_name(U)
    if level_name is None:
        level_name = utils.get_level_name(U)
    if time_name is None:
        time_name = utils.get_time_name(U)
    P0_pa = 1000
    LAT = U[lat_name][:]

比如上面怎么让UV为多维数据，而不是只是一个数据

Answer 1

UV是什么全靠你传进来个什么，你当它是多维的它就是多维的
当然IDE不会知道你心里以为的UV到底是什么类型，所以没法自动给你补全代码
一种办法是先给形参一个默认值，写成U=[[]]这种形式，那么它就是个二维list，你把代码都写好了之后再把默认值删掉