python通过函数判断选择读取哪一个文件夹中的文件
时间: 2024-02-12 12:02:54 浏览: 17
可以通过定义一个函数,来根据不同的条件选择读取哪一个文件夹中的文件。
首先定义一个函数,例如:
```
import os
def read_files(folder_path):
if folder_path == 'folder1':
files = os.listdir(folder_path)
for file in files:
file_path = os.path.join(folder_path, file)
if os.path.isfile(file_path):
with open(file_path, 'r') as f:
content = f.read()
print(content)
elif folder_path == 'folder2':
files = os.listdir(folder_path)
for file in files:
file_path = os.path.join(folder_path, file)
if os.path.isfile(file_path):
with open(file_path, 'r') as f:
content = f.read()
print(content)
else:
print('Invalid folder path')
```
这里以`folder1`和`folder2`为例,当函数参数为`folder1`时,读取`folder1`文件夹中的文件,当函数参数为`folder2`时,读取`folder2`文件夹中的文件,其他情况下输出错误提示信息。
接下来可以调用这个函数,例如:
```
read_files('folder1')
```
这里调用`read_files`函数,参数为`folder1`,将读取`folder1`文件夹中的文件。
完整的代码如下:
```
import os
def read_files(folder_path):
if folder_path == 'folder1':
files = os.listdir(folder_path)
for file in files:
file_path = os.path.join(folder_path, file)
if os.path.isfile(file_path):
with open(file_path, 'r') as f:
content = f.read()
print(content)
elif folder_path == 'folder2':
files = os.listdir(folder_path)
for file in files:
file_path = os.path.join(folder_path, file)
if os.path.isfile(file_path):
with open(file_path, 'r') as f:
content = f.read()
print(content)
else:
print('Invalid folder path')
read_files('folder1')
```
需要注意的是,在读取完成后需要关闭文件,可以使用`with open() as`语句来自动关闭文件。