python获取当前文件下的所有文件名

1、python获取当前文件下的所有文件名（txt）

import os

# os.listdir()方法获取文件夹名字，返回数组
def getAllFiles(targetDir):
    listFiles = os.listdir(targetDir)
    return listFiles

files = getAllFiles(r"D:\test")

# 写入list到txt文件中
with open(r"test.txt",'w+',encoding='utf-8')as f:
    # 列表生成式
    f.writelines(["'"+str(i)+"',\n" for i in files])

运行效果图

1.1、其他版本

# 导入os模块
import os

# path定义要获取的文件名称的目录
path = "D:\test"

# os.listdir()方法获取文件夹名字，返回数组
file_name_list = os.listdir(path)

# 转为转为字符串
file_name = str(file_name_list)

# replace替换"["、"]"、" "、"'"
file_name = file_name.replace("[", "").replace("]", "").replace("'", "").replace(",", "\n").replace(" ", "")

# 创建并打开文件list.txt
f = open(path + "\\" + "文件list.txt", "a")

# 将文件下名称写入到"文件list.txt"
f.write(file_name)

2、pthon文件的读取与写入（yaml）

from shutil import copyfile
from sys import exit
import re
import time
import os
import yaml

def getAllFiles(targetDir):
    files = []
    listFiles = os.listdir(targetDir)
    for i in range(0, len(listFiles)):
        path = os.path.join(targetDir, listFiles[i])
        if os.path.isdir(path):
            files.extend(getAllFiles(path))
        elif os.path.isfile(path):
            files.append(path)
    return files

files = getAllFiles(r"C:\Users\rc\Pictures\壁纸精选500张（四）")
a=files

# 写入list到yaml文件中
with open('ee.yaml','w+',encoding='utf-8')as f:
    # 写入yaml文件,加入allow_unicode=True参数防止写入的中文乱码
    yaml.dump(data=a, stream=f, allow_unicode=True)

# 读yaml文件，Loader=yaml.FullLoader更加安全
with open("ee.yaml",'r',encoding="utf-8") as f:
    data=yaml.load(stream=f, Loader=yaml.FullLoader)
    print(data, type(data))

    for i in range(0,len(data)):
        if i<=20:
            print((data[i]))
# print(data,type(data))