在Python中用@click.argument替换@click.option

Question

我有以下脚本，我使用 Python 中的 click 库中的 @click.option 。我想知道如何在不更改输出的情况下将其替换为 @click.argument。

import sys
import click
import pandas as pd
from sklearn import metrics
@click.command()

@click.option('--filename')
# @click.argument('result.csv', type=click.Path(exists=True))

def main(filename):

    df = pd.read_csv(filename)
    y_test = df["actual"].values
    y_pred = df["predicted"].values

    print('R Squared: ' +  str(metrics.r2_score(y_test, y_pred)))
    
if __name__ == '__main__':
    sys.argv = ['',  '--filename', 'result.csv']
    main()

Answer 1

import sys
import click
import pandas as pd
from sklearn import metrics
@click.command()

# @click.option('--filename')
## @click.argument('result.csv', type=click.Path(exists=True))
@click.argument('filename', type=click.Path(exists=True))
def main(filename):

    df = pd.read_csv(filename)
    y_test = df["actual"].values
    y_pred = df["predicted"].values

    print('R Squared: ' +  str(metrics.r2_score(y_test, y_pred)))
    
if __name__ == '__main__':
    # sys.argv = ['',  '--filename', 'result.csv']
    sys.argv = ['',  'result.csv']
    main()