在Python目录中搜索.txt，.scv文件

Question

我想在Python的给定目录中将所有扩展名为.txt和.scv的文件添加到列表中。 我当前的代码如下所示：

import glob
def core():
  old_path = '/home/demo/'
  files = glob.glob(old_path+(*.{txt, scv}))
  print(len(files))

但这不起作用。

Answer 1

您的代码中存在语法错误-通配符必须为字符串。 第二个问题是glob不支持花括号扩展（即，它不理解{txt, csv} ）。 因此，您需要使用基本的通配符：

import glob
def core():
    old_path = '/home/demo/'
    txt_files = glob.glob(old_path+'*.txt')
    csv_files = glob.glob(old_path+'*.csv')
    print(len(txt_files + csv_files))

core()