在Python中,你可以使用 `os` 模块来获取目录下的文件名。以下是获取目录下文件名的方法:
1. 使用 `os.listdir()` 函数获取指定目录下的所有文件和文件夹名称列表。
2. 使用 `os.path.isfile()` 函数过滤出列表中的文件。
3. 如果需要递归地读取目录下的所有文件名,可以使用 `os.walk()` 函数。
下面是一个示例代码,展示了如何使用这些方法:
```python
import os
def get_file_names(directory):
file_names = []
for filename in os.listdir(directory):
if os.path.isfile(os.path.join(directory, filename)):
file_names.append(filename)
return file_names
获取当前目录下的所有文件名
current_directory = os.getcwd()
files = get_file_names(current_directory)
print(files)
如果你需要获取特定类型的文件名,比如所有的 `.json` 文件,你可以修改 `get_file_names` 函数来过滤出这些文件:
```python
def get_file_names_with_suffix(directory, file_suffix):
file_names = []
for filename in os.listdir(directory):
if os.path.isfile(os.path.join(directory, filename)) and filename.endswith(file_suffix):
file_names.append(filename)
return file_names
获取当前目录下所有的.json文件名
json_files = get_file_names_with_suffix(current_directory, '.json')
print(json_files)
希望这能帮助你获取目录下的文件名