先上代码!!
第一部分
合成一个文件夹里所有的txt文件
import os
# 指定目录
directory ='D:\Signal_and_System_PBL\data-salt_pepper_noise'
# 获取目录下所有txt文件
txt_files = [f for f in os.listdir(directory) if f.endswith('.txt')]
# 合并所有txt文件的前四行
with open('D:\Signal_and_System_PBL\data-salt_pepper_noise\sum.txt', 'w') as outfile:
for fname in txt_files:
with open(os.path.join(directory, fname)) as infile:
for i, line in enumerate(infile):
if i < 4:
outfile.write(line)
else:
break
值得注意的是 directory后面是你放置需要合成的txt的文件夹地址。
open函数中是你要合成到的txt文件,这两个是有区别的,并且如果open函数后面的路径力已经包含了和最后合成的文件名称相同的文件,将会被直接覆盖而不是追加写入。
这段代码最后的for循环是因为我只需要读取txt文件中前四行代码,其余的代码不需要,所以添加了一个计数器,如果是要合成所有的数据的话,将for循环删掉添加一个下面的语句即可:
outfile.write(infile.read())
我只读前四个数据,我的txt示例如下:
合成完的结果如下:
第二部分
接下来是三段小功能,作用分别是将每一行中的等号替换成逗号、逗号替换成换行和让每一行行首顶格。
等号替换成逗号
# 定义文件路径
file_path = "D:\Signal_and_System_PBL\数据1.txt"
# 打开文件,读取内容
with open(file_path, "r") as f:
content = f.read()
# 用逗号替换等号
content = content.replace("=", ",")
# 重新打开文件,写入修改后的内容
with open(file_path, "w") as f:
f.write(content)
逗号替换成换行
# 打开文件并读取内容
with open(r'D:\Signal_and_System_PBL\data-salt_pepper_noise\sum.txt', 'r') as file:
content = file.read()
# 将逗号替换为换行符
content = content.replace(',', '\n')
# 将修改后的内容写入文件
with open(r'D:\Signal_and_System_PBL\data-salt_pepper_noise\sum.txt', 'w') as file:
file.write(content)
行首顶格
# 打开并读取内容
with open(r'D:\Signal_and_System_PBL\data-salt_pepper_noise\sum.txt', 'r') as file:
content = file.readlines()
# 去除每一行的空格和制表符
content = [line.strip() for line in content]
# 将修改后的内容写入文件
with open(r'D:\Signal_and_System_PBL\data-salt_pepper_noise\sum.txt', 'w') as file:
file.write('\n'.join(content))
效果是:
路径自己换吧,我懒得改了,这些要不要用看自己需求,这个图片的变化仅供参考
第三部分
这一部分是一个很简单的调用pandas库和matplotlib库来画表格
import pandas as pd
import matplotlib.pyplot as plt
# 读取文件
with open("D:\Signal_and_System_PBL\数据2.txt", "r") as f:
data = f.readlines()
# 处理数据
table_data = {}
for line in data:
line = line.strip()
# 如果该行不包含逗号,则过
if "," not in line:
continue
else:
key, value = line.split(",")
key = key.strip()
value = value.strip()
if key not in table_data:
table_data[key] = []
table_data[key].append(value)
# 制成表格
df = pd.DataFrame(table_data)
df.index = df.index + 1
# 显示表格
df.insert(0, 'Index', range(1, len(df)+1))
fig, ax = plt.subplots(figsize=(10,10))
ax.axis('off')
ax.axis('tight')
ax.table(cellText=df.values, colLabels=df.columns, loc='center')
plt.show()
值得注意的是,我这个代码中限定了只有一行中以逗号为分割,前面为标签,后面为数据的才会被用来画图,这个条件可以自己改,比如一行数据一行标签,必须以等号为分割等等等等。
运行结果就是这样的