python把文件夹名里的空格换成下划线

python把文件夹名里的空格换成下划线

import os

def rm_space():
    for f in os.listdir("."):
        if f.find(" ") > 0:
            n  =  f.replace(" ", "_")
            if n != f:
                os.rename(f, n)
            else:
                os.rename(f, "_"+n)
rm_space()