Это продолжение предыдущего вопроса. Я получил несколько хороших предложений по этому поводу, поэтому я решил попытать счастья еще раз.
from itertools import takewhile
if K is None:
illuminacond = lambda x: x.split(',')[0] != '[Controls]'
else:
illuminacond = lambda x: x.split(',')[0] != '[Controls]' and i < K
af=open('a')
bf=open('b', 'w')
cf=open('c', 'w')
i = 0
if K is None:
for line in takewhile(illuminacond, af):
line_split=line.split(',')
pid=line_split[1][0:3]
out = line_split[1] + ',' + line_split[2] + ',' + line_split[3][1] + line_split[3][3] + ',' \
+ line_split[15] + ',' + line_split[9] + ',' + line_split[10]
if pid!='cnv' and pid!='hCV' and pid!='cnv':
i = i+1
bf.write(out.strip('"')+'\n')
cf.write(line)
else:
for line in takewhile(illuminacond, af):
line_split=line.split(',')
pid=line_split[1][0:3]
out = line_split[1] + ',' + line_split[2] + ',' + line_split[3][1] + line_split[3][3] + ',' \
+ line_split[15] + ',' + line_split[9] + ',' + line_split[10]
if pid!='cnv' and pid!='hCV' and pid!='cnv':
i = i+1
bf.write(out.strip('"')+'\n')
Можно ли компактифицировать этот код? Если у меня есть что-то общее в двух таких циклах, очевидная возможность состоит в том, чтобы просто выделить общий код, но здесь, фу. Раздражает то, что единственная разница здесь — запись в c
.
Краткое описание кода: если K
не равно None, то перебрать K
строки a
и записать как b
, так и c
. В противном случае переберите все a
и просто запишите в b
.