lambda apply 多参数

import pandas as pd

def aid(arg1, arg2):
    return arg1 + arg2

def main() -> None:
    df = pd.DataFrame(
        [list(range(2))]*2
    )
    # print(df.apply(aid()))
    # TypeError: aid() missing 2 required positional arguments: 'arg1' and 'arg2'
    # print(df.apply(aid))
    # TypeError: aid() missing 1 required positional argument: 'arg2'
    print(df.apply(lambda x: aid(*x), axis=1))

if __name__ == '__main__':
    main()