关于[f for f in list1 if function(f) in list2]中，如果function(f)返回为空的问题

cao627

1. >>> [f for f in li if os.path.splitext(f)[1] in '.sh']
   
2. ['file', 'dd.sh']
   
3. >>> li = ['a.mp3','b.txt','cc.mp3','file']
   
4. >>> [f for f in li if os.path.splitext(f)[1] in '.mp3']
   
5. ['a.mp3', 'cc.mp3', 'file']
   

复制代码

os.path.splitext(f)[1]返回文件扩展名，如果文件没有扩展名返回空，这样的话就不会被指定扩展名过滤。如例子里的file。
怎么解决？

timespace

问题简单化：

1. [f for f in li if f.endswith('.sh')]

复制代码

如果有更复杂的过滤，还是写循环吧。

cao627

@timespace
谢谢！

Jpython

我觉得也是用 endwith

cao627

@Jpython
但这样只能简单过滤.
[f for f in li if os.path.splitext(f)[1] in '.mp3']可以用[f for f in li if f.endswith('.mp3')] 代替。
但 os.path.splitext(f)[1]会根据不同的文件f到多个扩展名，如果要取其中的两个以上，比如：
[f for f in li if os.path.splitext(f)[1] in ['.mp3','.sh']]
用 endwith就要用循环了。

timespace

回复 5# cao627
这要看具体情况，但恰好你的举例不用写循环。因为：

1. >>> '' in '.mp3' or '' in '.sh'
   
2. True
   
3. >>> '' in ('.mps', '.sh')
   
4. False

复制代码

cao627

@timespace
我也刚想到这个问题。上来看到你已经发现问题所在。
空字符串属于任何字符串，不属于任何不含空串的序列。我在一楼发帖时潜意识里将字符串'.mp3'等同['.mp3']了。