根据关键词匹配文件中特定的数值，然后绘制曲线图；
# 导远los数据的eph & ssr & los num曲线
import os
import matplotlib.pyplot as plt
import re
from PyQt5.QtWidgets import QApplication, QDialog, QVBoxLayout, QLabel, QLineEdit, QPushButton, QFileDialog
plt.rcParams["font.sans-serif"] = ["SimHei"] # 设置显示中文字体
plt.rcParams["axes.unicode_minus"] = False # 设置正常显示符号
# os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
# os.environ["CUDA_VISIBLE_DEVICES"] = "1"
class MyDialog(QDialog):
def __init__(self, parent=None):
super(MyDialog, self).__init__(parent)
self.setWindowTitle('nLos_绘图工具')
self.setGeometry(100, 100, 200, 200)
# 创建垂直布局管理器
layout = QVBoxLayout()
# 创建文件选择框
label3 = QLabel('请点击浏览按钮，选择nLos的log文件；\n再点击执行按钮，绘制曲线。', self)
layout.addWidget(label3)
self.file_input1 = QPushButton('浏览...', self)
layout.addWidget(self.file_input1)
self.file_input1.clicked.connect(self.open_file_dialog1)
# 添加确认执行按钮
self.ok_button = QPushButton('执行', self)
self.ok_button.clicked.connect(self.execute)
layout.addWidget(self.ok_button)
# 将布局设置为主窗口的布局
self.setLayout(layout)
def open_file_dialog1(self):
file_dialog = QFileDialog(self)
file_dialog.setFileMode(QFileDialog.ExistingFiles)
if file_dialog.exec_():
file_paths = file_dialog.selectedFiles()
merged_filename = ""
if len(file_paths) > 1:
# 获取第一个文件的目录和文件名，用于创建合并后的文件名
directory = os.path.dirname(file_paths[0])
filename = os.path.splitext(os.path.basename(file_paths[0]))[0]
merged_filename = os.path.join(directory, f'{filename}_merged').replace('\\', '/')
with open(merged_filename, 'w') as merged_file:
for path in file_paths:
with open(path, 'r') as file:
merged_file.write(file.read())
merged_file.write('\n') # 在每个文件之间添加换行符
elif len(file_paths) == 1:
merged_filename = file_paths
else:
print('Please select files.')
if merged_filename:
self.file_input1.setText(''.join(merged_filename))
def execute(self):
# 执行操作，可以在这里处理输入框和文件选择框的内容
file_path_s = self.file_input1.text()
# 匹配的关键字
keyword_s = 'start time'
keyword_i = 'input.time'
keyword_e = 'eph num'
keyword_d = 'dpi_sdk ssr num'
keyword_l = 'nLos'
with open(file_path_s, 'r', encoding="utf-8", errors="ignore") as file:
lines = file.readlines()
it_list, en_list, dn_list, ln_list = [], [], [], []
it = 0
for index, line in enumerate(lines):
# 匹配相近的两行关键词
if line.startswith(keyword_e) and lines[index + 1].startswith(keyword_d):
# 空格分隔，去掉末尾的换行
en = line.split(' ')[-1].strip('\n')
dn = lines[index + 1].split(' ')[-1].strip('\n')
# 解决绘图时，y轴排序错乱问题，字符转换成整型
en_list.append(int(en))
dn_list.append(int(dn))
# 匹配nLos关键词
if keyword_l in line:
# 空格分隔，去掉末尾的换行
ln = re.split("[,| |:|=]+", line)[4]
# print(ln)
it = re.split("[,| |:|=]+", line)[-1].strip('\n')
# print(it)
ln_list.append(int(ln))
it_list.append(it)
# 剔除掉第一个
# ln_list.pop(0)
# print(len(ln_list))
# 添加标签和标题
plt.xlabel('时间')
plt.ylabel('数值')
# plt.title(keyword_e + ' & ' + keyword_d + ' & ' + keyword_l, fontsize=24)
file_name = os.path.basename(file_path_s)
plt.title(file_name, fontsize=24)
# 显示网格线和设置背景色
plt.grid(True)
plt.gca().set_facecolor('lightgrey')
# plt.plot(it_list, en_list, it_list, dn_list)
# linestyle = "--" 表示虚线，“：”表示点虚线，"-."表示点划线，
plt.plot(it_list, en_list, linestyle='--', label=keyword_e, marker='o')
plt.plot(it_list, dn_list, linestyle=':', label=keyword_d, marker='v')
plt.plot(it_list, ln_list, linestyle='-.', label=keyword_l, marker=',')
# 添加图例
plt.legend()
# 显示图表
plt.show()
if __name__ == '__main__':
import sys
app = QApplication(sys.argv)
dialog = MyDialog()
# 解决子窗口不能操作的问题
dialog.show()
dialog.exec_()
文件中的关键词发生变化，调整后：
# 导远los数据的eph & ssr & los num曲线
import os
import matplotlib.pyplot as plt
from matplotlib.dates import AutoDateLocator
import re
from PyQt5.QtWidgets import QApplication, QDialog, QVBoxLayout, QLabel, QLineEdit, QPushButton, QFileDialog
plt.rcParams["font.sans-serif"] = ["SimHei"] # 设置显示中文字体
plt.rcParams["axes.unicode_minus"] = False # 设置正常显示符号
# os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
# os.environ["CUDA_VISIBLE_DEVICES"] = "1"
class MyDialog(QDialog):
def __init__(self, parent=None):
super(MyDialog, self).__init__(parent)
self.setWindowTitle('los绘图工具')
self.setGeometry(100, 100, 200, 200)
# 创建垂直布局管理器
layout = QVBoxLayout()
# 创建文件选择框
label3 = QLabel('请点击浏览按钮，选择los的log文件；\n再点击执行按钮，绘制曲线。\n兼容导远V1.2-V1.4版本', self)
layout.addWidget(label3)
self.file_input1 = QPushButton('浏览...', self)
layout.addWidget(self.file_input1)
self.file_input1.clicked.connect(self.open_file_dialog1)
# 添加确认执行按钮
self.ok_button = QPushButton('执行', self)
self.ok_button.clicked.connect(self.execute)
layout.addWidget(self.ok_button)
# 将布局设置为主窗口的布局
self.setLayout(layout)
def open_file_dialog1(self):
file_dialog = QFileDialog(self)
file_dialog.setFileMode(QFileDialog.ExistingFiles)
if file_dialog.exec_():
file_paths = file_dialog.selectedFiles()
merged_filename = ""
if len(file_paths) > 1:
# 获取第一个文件的目录和文件名，用于创建合并后的文件名
directory = os.path.dirname(file_paths[0])
filename = os.path.splitext(os.path.basename(file_paths[0]))[0]
merged_filename = os.path.join(directory, f'{filename}_merged').replace('\\', '/')
with open(merged_filename, 'w') as merged_file:
for path in file_paths:
with open(path, 'r') as file:
merged_file.write(file.read())
merged_file.write('\n') # 在每个文件之间添加换行符
elif len(file_paths) == 1:
merged_filename = file_paths
else:
print('Please select files.')
if merged_filename:
self.file_input1.setText(''.join(merged_filename))
def execute(self):
# 执行操作，可以在这里处理输入框和文件选择框的内容
file_path_s = self.file_input1.text()
# 匹配的关键字
keyword_s = 'start' # 新旧版本的start与time之间的空格不一致
keyword_i = 'input.time'
keyword_e = 'eph num'
keyword_d = 'dpi_sdk ssr num'
keyword_l = 'los num'
keyword_t = 'time'
with open(file_path_s, 'r', encoding="utf-8", errors="ignore") as file:
lines = file.readlines()
it_list, en_list, dn_list, ln_list = [], [], [], []
it = 0
for index, line in enumerate(lines):
# 匹配相近的三行关键词
if line.startswith(keyword_s) and lines[index + 1].startswith(keyword_e) and lines[
index + 2].startswith(keyword_d):
# 空格分隔，去掉末尾的换行
# it = line.split(' ')[-1].strip('\n')
en = lines[index + 1].split(' ')[-1].strip('\n')
dn = lines[index + 2].split(' ')[-1].strip('\n')
# print(it, en, dn)
# it_list.append(it)
# 解决绘图时，y轴排序错乱问题，字符转换成整型
en_list.append(int(en))
dn_list.append(int(dn))
# 匹配los num关键词
if keyword_l in line and lines[index + 1].startswith(keyword_t):
# 空格分隔，去掉末尾的换行
ln = line.split(' ')[-1].strip('\n')
ln_list.append(int(ln))
it = re.split("[,| ]+", lines[index + 1])[1].strip('\n')
it_list.append(it)
# 剔除掉第一个，数量需要相等，不然报错
# ln_list.pop(0)
# print(len(ln_list))
# print(len(it_list), len(ln_list), len(en_list), len(dn_list))
# 列表的数量需要相等，不然绘图的时候会报错，切片获取最小长度的列表
if len(it_list) > len(en_list):
it_list = it_list[:len(en_list)]
ln_list = ln_list[:len(en_list)]
elif len(it_list) < len(en_list):
en_list = en_list[:len(it_list)]
dn_list = dn_list[:len(it_list)]
# 添加标签和标题
plt.xlabel('GPST时间')
plt.ylabel('num数值')
# plt.title(keyword_e + ' & ' + keyword_d + ' & ' + keyword_l, fontsize=24)
file_name = os.path.basename(file_path_s)
plt.title(file_name, fontsize=24)
# 自动调整刻度
plt.gca().xaxis.set_major_locator(AutoDateLocator())
# 显示网格线和设置背景色
plt.grid(True)
plt.gca().set_facecolor('lightgrey')
# plt.plot(it_list, en_list, it_list, dn_list)
# linestyle = "--" 表示虚线，"："表示点虚线，"-."表示点划线，
plt.plot(it_list, en_list, linestyle='--', alpha=0.5, label=keyword_e, marker='o', linewidth=4)
plt.plot(it_list, dn_list, linestyle=':', alpha=0.5, label=keyword_d, marker='v', linewidth=1)
plt.plot(it_list, ln_list, linestyle='-.', label=keyword_l, marker=',')
# 添加图例
plt.legend()
# 显示图表
plt.show()
if __name__ == '__main__':
import sys
app = QApplication(sys.argv)
dialog = MyDialog()
# 解决子窗口不能操作的问题
dialog.show()
dialog.exec_()
扫一扫
604
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
