如何运行程序，如 ./main.py --fields column1, column2答案

【问题标题】：How to run the program like ./main.py --fields column1, column2如何运行程序，如 ./main.py --fields column1, column2
【发布时间】：2023-02-02 00:11:29
【问题描述】：

我有

df = pd.read_csv(csv_raw)
data_json = df.to_json(orient='records')

在df中，用户可以通过运行代码./main.py --fields column1, column2在终端中选择不同的列。我知道我应该使用 argparse 库。但是我尝试过的方法并不值得我付出努力。

我试过下面的代码，但它对我不起作用：

parser = argparse.ArgumentParser(description='Process rows to display.')
parser.add_argument('--fields', type=str, help='Rows to display')

args = parser.parse_args()
select = args.select.split(',') if args.select else None

df 包含：

date                 column1  column2     column3   column4 
0   2022-01-01          1    2.550000     Unknown  facebook
1   2022-01-01          5  2.470000  Unknown  facebook
2   2022-01-01          2  2.620000  Unknown  facebook
3   2022-01-01          3  2.480000  Unknown  facebook
4   2022-01-01          2  2.440000  Unknown  facebook

【问题讨论】：

标签： python

【解决方案1】：

import sys

print(sys.argv)

if len(sys.argv) > 1:
    select = sys.argv[1:]
else:
    None

【讨论】：