0

ファイル名のリストからファイルの種類を除外するフィルター関数を作成しました。

>>> l1
['180px-Cricketball.png', 'AgentVinod_450.jpg', 'Cricketball.bmp', 'Django-1.4', 'Django-1.4.tar.gz', 'Firefox Setup 11.0.exe', 'I-Will-Do-The-Talking-Tonight-(Muskurahat.Com).mp3', 'kahaani-.jpg', 'Never gonna leave this bed.mp3', 'Piya-Tu-Kaahe-Rootha-Re-(Muskurahat.Com).mp3', 'pygame-1.9.1release', 'pygame-1.9.1release.zip', 'pygame-1.9.2a0.win32-py2.7.msi', 'python-2.7.2.msi', 'python-3.1.2.msi', 'Resume.doc', 'selenium-2.20.0', 'selenium-2.20.0.tar.gz', 'sqlite-shell-win32-x86-3071100.zip', 'wxdesign_220a.exe', 'YTDSetup.exe']
>>> def myfilt(subject):
    if re.search('.jpg',subject):
        return True


>>> filter(myfilt,l1)
['AgentVinod_450.jpg', 'kahaani-.jpg']

これはうまくいきます。

ここで、これをより柔軟にしたいとします。ファイルタイプを関数に渡したいです。だから私は関数を書き直します

>>> def myfilt(subject,filetype):
    if re.search(filetype,subject):
        return True

フィルター関数を介してファイルタイプを渡すにはどうすればよいですか?

私は試した：

>>> filter(myfilt(l1,filetype),l1)

Traceback (most recent call last):
  File "<pyshell#32>", line 1, in <module>
    filter(myfilt(l1,filetype),l1)
  File "<pyshell#28>", line 2, in myfilt
    if re.search(filetype,subject):
  File "C:\Python27\lib\re.py", line 142, in search
    return _compile(pattern, flags).search(string)
TypeError: expected string or buffer

何も機能しません。何か案は？

4

1 に答える 1

10

filter()そのような場合ではなく、通常はリスト内包表記を使用します。

[x for x in l1 if myfilt(x, filetype)]

本当に使用したい場合はfilter()、ラムダ関数を使用できます

filter(lambda x: myfilt(x, filetype), l1)

またはfunctools.partial():

filter(functools.partial(myfilt, filetype=filetype), l1)

ただし、リスト内包表記は最も簡単で読みやすいオプションのようです。

于 2012-04-10T12:35:09.130 に答える