在Python中将文件名写入数组

import os

file_names = []

directory = '/path/to/directory'  # 替换为实际的目录路径
for filename in os.listdir(directory):
    file_names.append(filename)

for filename in os.listdir(directory):
    if os.path.splitext(filename)[1] == '.txt':  # 替换为实际的文件扩展名
        file_names.append(filename)

import os

directory = '/path/to/directory'  # 替换为实际的目录路径
file_names = []

for filename in os.listdir(directory):
    file_names.append(filename)

print(file_names)