pycファイルを削除する

こんばんは。Aliceです。
先ほど、インタプリタで「.py」ファイルをimportすると出る「.pyc」ファイルを削除するスクリプトを書いてみました。
ソースコードはこちらです。


#!/usr/bin/python
#coding:utf-8

import os

flist = os.listdir(os.getcwd())
for i in range(len(flist)):
root = flist[i].split(".")
ext = root[1]
if ( ext == "pyc" ):
cmd = "rm "+str(flist[i])
os.system(cmd)
print "the end."

おおまかになにをしているかというと、まずOSモジュールでディレクトリ下のファイルをリストで出力して、
その後、ループを書いて、リスト一つ一つをドットで区切り、区切ったリストの1番目の要素が「pyc」(=拡張子)ならば、rmコマンドでファイルを削除する、といったものです。

もし、ディレクトリ下のpycが邪魔になったら、このスクリプトを使ってみてください。


** ちなみに **

このプログラムは、bashで書くとこうなります、多分。


$ files=`ls | grep .pyc`
$ rm $files
これをPythonに応用すると、

>>> cmd = "ls | grep .pyc"
>>> files = os.system(cmd)
****.pyc
****.pyc
>>> files
0
となってしまい、変数を使っても0しか代入されません。
良い方法があれば、ぜひ教えてください。
以上、Aliceでした!