我有这样的东西:
extensionsToCheck = ['.pdf', '.doc', '.xls']
for extension in extensionsToCheck:
if extension in url_string:
print(url_string)
我想知道在Python中更优雅的方法是什么(不使用for循环)?我在想类似这样的东西(比如从C/ c++中)，但它不起作用:
if ('.pdf' or '.doc' or '.xls') in url_string:
print(url_string)
编辑:我有点被迫解释这与下面被标记为潜在重复的问题有什么不同(所以我猜它不会关闭)。
区别在于，我想检查字符串是否是某个字符串列表的一部分，而另一个问题是检查字符串列表中的字符串是否是另一个字符串的子字符串。相似，但不完全相同，当你在网上寻找答案时，语义很重要。这两个问题实际上是要解决彼此相反的问题。但两者的解决方案是一样的。
当前回答
这是我能想到的最简单的方法。
list_ = ('.doc', '.txt', '.pdf')
string = 'file.txt'
func = lambda list_, string: any(filter(lambda x: x in string, list_))
func(list_, string)
# Output: True
同样，如果有人需要保存字符串中的元素，他们可以使用这样的方法:
list_ = ('.doc', '.txt', '.pdf')
string = 'file.txt'
func = lambda list_, string: tuple(filter(lambda x: x in string, list_))
func(list_, string)
# Output: '.txt'
其他回答
extensionsToCheck = ('.pdf', '.doc', '.xls')
'test.doc'.endswith(extensionsToCheck) # returns True
'test.jpg'.endswith(extensionsToCheck) # returns False
这是我能想到的最简单的方法。
list_ = ('.doc', '.txt', '.pdf')
string = 'file.txt'
func = lambda list_, string: any(filter(lambda x: x in string, list_))
func(list_, string)
# Output: True
同样，如果有人需要保存字符串中的元素，他们可以使用这样的方法:
list_ = ('.doc', '.txt', '.pdf')
string = 'file.txt'
func = lambda list_, string: tuple(filter(lambda x: x in string, list_))
func(list_, string)
# Output: '.txt'
为了防止有人再次面临这样的任务，这里有另一个解决方案:
extensionsToCheck = ['.pdf', '.doc', '.xls']
url_string = 'file.doc'
res = [ele for ele in extensionsToCheck if(ele in url_string)]
print(bool(res))
> True
如果需要单行解决方案，请使用列表推导。下面的代码在包含。doc， .pdf和.xls扩展名时返回一个包含url_string的列表，在不包含扩展名时返回空列表。
print [url_string for extension in extensionsToCheck if(extension in url_string)]
注意:这只是为了检查它是否包含，当想要提取与扩展名匹配的确切单词时，这是没有用的。
最好正确地解析URL -这样你就可以正确地处理http://.../file.doc?foo和http://.../foo.doc/file.exe。
from urlparse import urlparse
import os
path = urlparse(url_string).path
ext = os.path.splitext(path)[1]
if ext in extensionsToCheck:
print(url_string)
