Управляйте выводом цикла for
Мне нужно сравнить следующие элементы списка sequences:
sequences = ['sphere_v002_', 'sphere_v002_0240_', 'test_single_abc_f401']
к:
folder = 'sphere_v002'
а затем поработайте над элементами списка, содержащими folder.
У меня для этого есть рабочая функция, но я хочу ее улучшить.
Текущий код:
foundSeq = False
for seq in sequences:
headName = os.path.splitext(seq.head())[0]
#Check name added exception for when name has a last underscore
if headName == folder or headName[:-1] == folder:
foundSeq = True
sequence = seq
if not foundSeq:
...
Мое улучшение выглядит так:
if any(folder in os.path.splitext(seq.head())[0] for seq in sequences):
print seq
Но тогда я получаю следующую ошибку:
local variable seq referenced before the assignment
Как я могу получить правильный результат, работая с улучшенным решением?
Ответы 2
any возвращает только логическое значение, он не будет сохранять в переменной seq элемент в sequences, когда ваше условие удовлетворяется.
Что вы можете сделать, так это использовать генератор и использовать тот факт, что None является «ложным»:
def get_seq(sequences, folder):
for seq in sequences:
if folder in os.path.splitext(seq.head())[0]:
yield seq
for seq in get_seq(sequences, folder):
print seq
Вы можете переписать это, если хотите, как выражение генератора:
for seq in (i for i in sequences if folder in os.path.splitext(i.head())[0]):
print seq
Если условие никогда не указано, генератор или выражение генератора не выдаст никаких значений, и логика в вашем цикле не будет обработана.
Как указывает jpp, any просто возвращает логическое значение. Так что если есть, то это не лучшее решение в данном конкретном случае.
Как и предположил thebjorn, наиболее эффективный код для нас пока состоит в использовании функции filter.
sequences = ['sphere_v002_', 'sphere_v002_0240_', 'test_single_abc_f401']
match = filter(lambda x: 'sphere_v002' == x[:-1] or 'sphere_v002' == x, sequences)
print match
['sphere_v002_']
Может, воспользуетесь функцией
filter..?