Python os.listdir方法(获取文件夹目录下的内容)

一、os.listdir方法

• os.listdir方法。参数为文件夹路径, 可以返回文件夹下的所有子文件、文件名称, 但不能返回子文件夹下的文件

import os

path = r"C:\Users\admin\Desktop\主文件夹"

# 获取文件夹下所有内容，不包括子文件夹下的内容，已列表的形式返回
print(os.listdir(path))

# 方法一：获取后缀为“.txt”的文件
for i in os.listdir(path):
    if os.path.splitext(i)[1] == '.txt':
        print(i)

# 方法二：获取后缀为“.txt和.sh”的文件
for i in os.listdir(path):
    if i.endswith(('.txt', '.sh')):
        print(i)