通常,如果我必须在多个文件之间执行重叠,我只需执行:
python -c 'import sys;print "".join(sorted(set.intersection(*[set(open(a).readlines()) for a in sys.argv[1:]])))' File1 File2 File3 File4 ....
但是,如果我必须通过解析器获取它,我该怎么做?
围绕这个脚本,一个简短的例子就足够了,之后我可以玩了。实际上,我希望用户给出两个输入:要对其执行 ioperation 的文件和此处的操作是交集。
请帮忙,而不是投票关闭帖子。
我想我必须展示我在做什么,这里有一个例子:
usage = "Usage: %prog -i <file1>[,<file2>,...] -o <out> -ops "
version = "%prog " + str(VERSION)
parser = OptionParser(version=version, usage=usage)
group1 = OptionGroup(parser, 'Optional')
parser.add_option("-i",
dest="featurefile",
help="Gene list",
metavar="FILE")
parser.add_option("-o",
dest="outfile",
help="output file (type determined by extension)",
metavar="FILE")
# Optional arguments
group1.add_option("--op",
dest="operation",
help="Intersection, Union & Uniques (default Intersection)",
default=None,
type="string")
parser.add_option_group(group1)
(options, args) = parser.parse_args()
现在这个集合是解析器的事情(完成),现在第二步是获取操作(交集)和文件并执行它们。在这一步,我迷路了。
谢谢
由于您使用的是 Python 2.7,请使用 argparse 模块而不是旧optparse
:
import argparse
import itertools as IT
VERSION = 0.1
version = "%(prog)s " + str(VERSION)
operation = {
'intersection': set.intersection,
'union': set.union,
'uniques': set.difference}
parser = argparse.ArgumentParser()
parser.add_argument('--version', action='version', version=version)
parser.add_argument('-i', '--featurefiles', nargs='+', help='Gene list',
metavar='FILE')
parser.add_argument('-o', '--outfile', help="output file (type determined by extension)",
metavar='FILE')
parser.add_argument('--op', dest="operation",
help="Intersection, Union & Uniques (default Intersection)",
default='intersection',
# can't use type=operation.get. See http://bugs.python.org/issue16516
type=lambda x: operation[x])
args = parser.parse_args()
print(args)
if args.outfile:
with open(args.outfile, 'w') as f:
for group_size in (len(args.featurefiles), 2):
for group in IT.combinations(args.featurefiles, group_size):
result = ''.join(sorted(reduce(
args.operation,
(set(open(filename))
for filename in group))))
f.write('{g}:n{r}n'.format(g=str(group), r=result))
您可以从 CLI 调用脚本,如下所示:
script.py -i file1 file2 file3 --op uniques -o /tmp/out
对上面使用的工具的引用:
- 参数解析
- itertools.types
- 减少
- 与语句
- str 格式方法