问题描述
我有这个列表理解:
[''.join(letter for letter in word if not letter.isdigit()) for word in words]
我想要做的是对列表 word
中的每个 words
检查 word
是否包含 数字。如果单词有数字字符,它将从最终输出中排除。
例如,remove_numbers(['lion','tiger99','yes7','9'])
给我:
['lion','tiger','yes','']
['lion','yes']
我尝试在列表推导式中添加 if not word
。不幸的是,这不起作用。有人可以帮忙吗?
解决方法
if not word
在您的示例中不起作用,因为最后一个 "word" 不是空字符串 ''
但它是 '9'
,您的 processed output 是空字符串,它被附加到您的输出数组中。
@aminrd 的建议应该这样做,这里我们检查输入的单词是否是纯数字,但这不适用于小数。
示例代码:
words = ['lion','tiger300','yes7','9']
print([''.join(letter for letter in word if not letter.isdigit()) for word in words if not word.isnumeric()])
# or using regex
import re
print([re.sub('\d','',word) for word in words if not word.isnumeric()])
输出
['lion','tiger','yes']
既然你想知道如何在list comprehension中检查输出词是否为空
代码:
words = ['lion','9']
print([x for x in [''.join(letter for letter in word if not letter.isdigit()) for word in words] if x])
输出
['lion','yes']
但是这种方法效率低下,应该首选第一种方法,因为它只创建一个列表,这会创建两个列表,第一个更易于阅读。