Python-从当前文件夹读取文件

Question

我用 matplotlib 编写了一些拟合查看器。现在我有一个问题。当我开始编写脚本时，我不需要从当前文件夹中读取后缀为 .fts 的数据文件。例如

import pyfits
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.widgets import Button


image_list=['sun0001.fts','sun0002.fts','sun0003.fts','sun0004.fts','sun0005.fts','sun0006.fts','sun0007.fts','Moon0005.fts']

image_list是我手动制作的。但现在我想使用一些脚本自动从文件夹中读取文件，如下所示。

感谢解答

Answer 1

使用glob模块，例如，如果我想搜索当前目录中的JPEG文件:

In [180]: from glob import glob

In [181]: glob('*.jpg')
Out[181]: 
['IMG_20130626_193201.jpg',
 'IMG_20130926_191134.jpg',
 'IMG_20130926_191143.jpg',
 'IMG_20130926_191157.jpg']