Python: Passer des fonctions avec des arguments à une fonction intégrée?

Question

comme ça question Je veux passer une fonction avec des arguments. Mais je veux passer aux fonctions intégrées.

Exemple:

files = [ 'hey.txt', 'hello.txt', 'goodbye.jpg', 'howdy.gif' ] 

def filterex(path, ex): 
    pat = r'.+\.(' + ex + ')$' 
    match = re.search(pat, path)   
    return match and match.group(1) == ex) 

je pouvais utiliser ce code avec une boucle et une instruction if, mais il est plus court et peut-être plus lisible à utiliser un filtre (func, seq). Mais si je comprends bien, la fonction que vous utilisez avec le filtre prend seulement un argument qui est l'élément de la séquence.

Alors je me demandais s'il était possible de passer plus d'arguments?

Answer 1

def make_filter(ex): 
    def do_filter(path): 
     pat = r'.+\.(' + ex + ')$' 
     match = re.search(pat, path) 
     return match and match.group(1) == ex 
    return do_filter 

filter(make_filter('txt'), files) 

Ou si vous ne voulez pas modifier filterex:

filter(lambda path: filterex(path, 'txt'), files) 

Vous pouvez utiliser une compréhension de la liste, comme le suggère gnibbler:

[path for path in files if filterex(path, 'txt')] 

Vous pouvez également utiliser un générateur compréhension, ce qui pourrait être particulièrement utile si vous aviez une grande liste:

(path for path in files if filterex(path, 'txt')) 

Answer 2

Voici une compréhension de liste qui fait la même chose

import os 
[f for f in files if os.path.splitext(f)[1]=="."+ex]