關于[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')]

復制代碼

如果有更復雜的過濾，還是寫循環(huán)吧。

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就要用循環(huán)了。

timespace

回復 5# cao627
這要看具體情況，但恰好你的舉例不用寫循環(huán)。因為：

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

復制代碼

cao627

@timespace
我也剛想到這個問題。上來看到你已經發(fā)現問題所在。
空字符串屬于任何字符串，不屬于任何不含空串的序列。我在一樓發(fā)帖時潛意識里將字符串'.mp3'等同['.mp3']了。