Question

我有一个工作代码来打印csv列中的随机行。

#!/usr/bin/python

import csv   

import random 

**col**=2

with open('<filename>','r') as f:
        reader=csv.reader(f)
        data=[row[col] for row in reader]
    from random import shuffle
    shuffle(data)
    print '\n'.join(data[:**100**])
    f.close();

我想对此脚本进行参数化。通过，位置（col）＆amp;样本（例如100个值）

我无法找到一个可以解释这一点的体面教程。有什么想法吗？

Answer 1

您可以使用这样的sys模块将命令行参数传递给python脚本。

import sys

name_of_script = sys.argv[0]
position = sys.argv[1]
sample = sys.argv[2]

然后你的命令行将是......

./ myscript.py 10 100

Answer 2

使用argparse模块：

argparse模块可以轻松编写用户友好的命令行接口。该程序定义了它需要的参数，以及 argparse将弄清楚如何解析sys.argv中的那些。该 argparse模块还会自动生成帮助和使用消息当用户给程序提供无效参数时会发出错误。

它功能非常强大：您可以指定帮助消息，进行验证，提供默认值。无论您能想到如何使用命令行参数。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("-p", "--position", type=int)
parser.add_argument("-s", "--sample", type=int)

args = parser.parse_args()
col = args.position
sample = args.sample

print col
print sample

这是命令行上的内容：

$ python test.py --help
usage: test.py [-h] [-p POSITION] [-s SAMPLE]

optional arguments:
  -h, --help            show this help message and exit
  -p POSITION, --position POSITION
  -s SAMPLE, --sample SAMPLE

$ python test.py -p 10 -s 100
10
100
$ python test.py --position 10 --sample 100
10
100

说到您提供的代码：

unused import random声明
将from random import shuffle移至脚本顶部
无需致电f.close()（特别是;） - with处理自动关闭文件

以下是修复后代码的外观：

#!/usr/bin/python
import argparse
import csv
from random import shuffle


parser = argparse.ArgumentParser()
parser.add_argument("-p", "--position", type=int)
parser.add_argument("-s", "--sample", type=int)

args = parser.parse_args()

with open('<filename>', 'r') as f:
    reader = csv.reader(f)
    data = [row[args.position] for row in reader]
    shuffle(data)
    print '\n'.join(data[:args.sample])

Answer 3

谢谢你们。

由于缺乏许可，我无法安装argaprse - 同时。

这是解决方案 -

#!/usr/bin/python

import csv   # This will help us reading csv formated files.
import sys
import optparse #import optparse

from random import shuffle

parser=optparse.OptionParser()

# import options

parser.add_option('-f','--filename',help='Pass the csv filename')
parser.add_option('-p','--position',help='column position in the file',type=int)
parser.add_option('-s','--sample',help='sample size',type=int)

(opts,args) = parser.parse_args() # instantiate parser


# Program to select random values

with open('<filepath>'+opts.filename,'r') as f:
        reader=csv.reader(f)
        data=[row[opts.position] for row in reader]

        shuffle(data)

        #print '\n'.join(data[:opts.sample])

# create o/p file
file=open("<opfilename>.txt","w")
file.write('\n'.join(data[:opts.sample]))
file.close()

Python：将参数传递给脚本

3 个答案: