-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path连接1.txt
49 lines (42 loc) · 2.1 KB
/
连接1.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
》》使用python编写程序实现以下步骤:
1.当前目录下的summarize文件夹中的txt文件使用数字命名。用utf-8编码读取当前目录下的summarize文件夹中的txt文件并从小到大排序
2.第1步排序后每20个文件连接为1个文件,每个文件内容开头添加用带中括号的对应数字(例如 3.txt 的内容前面添加“【3】”),每个文件内容末尾添加换行。
3.把第二步得到的文件用utf-8编码保存到当前目录的“chapter2”文件夹,第一个文件命名为1.txt ,后面每次保存文件名数字加1。
》
```python
import os
import re
def read_and_sort_files(directory):
files = [f for f in os.listdir(directory) if f.endswith('.txt')]
files.sort(key=lambda x: int(re.search(r'\d+', x).group()))
contents = []
for file in files:
with open(os.path.join(directory, file), 'r', encoding='utf-8') as f:
content = f.read()
contents.append((file, content))
return contents
def concatenate_files(contents, batch_size):
concatenated_contents = []
for i in range(0, len(contents), batch_size):
batch = contents[i:i + batch_size]
concatenated_content = ""
for file, content in batch:
number = re.search(r'\d+', file).group()
concatenated_content += f"【{number}】{content}\n"
concatenated_contents.append(concatenated_content)
return concatenated_contents
def save_concatenated_files(concatenated_contents, output_directory):
if not os.path.exists(output_directory):
os.makedirs(output_directory)
for i, content in enumerate(concatenated_contents):
with open(os.path.join(output_directory, f"{i+1}.txt"), 'w', encoding='utf-8') as f:
f.write(content)
def main():
input_directory = 'summarize'
output_directory = 'chapter2'
batch_size = 20
contents = read_and_sort_files(input_directory)
concatenated_contents = concatenate_files(contents, batch_size)
save_concatenated_files(concatenated_contents, output_directory)
main()
```