【Python学习】读取文件夹下的所有文件

代码

# 读取目录下所有内容：当前路径下所有非目录子文件
# 返回文件名 []
def getallfiles(file_dir):
    for root,dirs,files in os.walk(file_dir):
        print('files:{}'.format(files))
        return  files

	
files=getallfiles('./')
file_txt=[]
for i in range(len(files)):
    file=files[i].split('.')  # ['data1', 'txt']

    if file[1]!='txt':
        continue
    print(file)
    file_txt.append(file[0])
print(file_txt)
print(len(file_txt)

输出

files:['data10_cat.txt', 'data11_cat.txt', 'data12_cat.txt', 'data13_cat.txt', 'data14_cat.txt', 'data15_cat.txt', 'data1_cat.txt', 'data2_cat.txt', 'data3_cat.txt', 'data4_cat.txt', 'data5_cat.txt', 'data6_cat.txt', 'data7_cat.txt', 'data8+9_cat.txt', 'normalize.py']
['data10_cat', 'txt']
['data11_cat', 'txt']
['data12_cat', 'txt']
['data13_cat', 'txt']
['data14_cat', 'txt']
['data15_cat', 'txt']
['data1_cat', 'txt']
['data2_cat', 'txt']
['data3_cat', 'txt']
['data4_cat', 'txt']
['data5_cat', 'txt']
['data6_cat', 'txt']
['data7_cat', 'txt']
['data8+9_cat', 'txt']
['data10_cat', 'data11_cat', 'data12_cat', 'data13_cat', 'data14_cat', 'data15_cat', 'data1_cat', 'data2_cat', 'data3_cat', 'data4_cat', 'data5_cat', 'data6_cat', 'data7_cat', 'data8+9_cat']
14