自定义函数sdwu_get_content(),分别读取data文件夹下的neg和pos文件夹中的文件。
时间: 2024-05-10 16:18:17 浏览: 5
以下是示例代码:
```python
import os
def sdwu_get_content(folder_path):
contents = []
for file_name in os.listdir(folder_path):
file_path = os.path.join(folder_path, file_name)
with open(file_path, 'r', encoding='utf-8') as f:
content = f.read()
contents.append(content)
return contents
neg_folder_path = './data/neg'
pos_folder_path = './data/pos'
neg_contents = sdwu_get_content(neg_folder_path)
pos_contents = sdwu_get_content(pos_folder_path)
```
此函数接收一个文件夹路径作为参数,遍历文件夹中的所有文件,并读取每个文件的内容。最后返回一个包含所有文件内容的列表。在上述示例中,我们分别调用该函数读取了neg文件夹和pos文件夹中的所有文件内容,并将其存储在了neg_contents和pos_contents两个列表中。