优化SA打标界面前景色设置逻辑,简化条件判断
This commit is contained in:
@ -526,41 +526,39 @@ class DataFrameModel(QAbstractTableModel):
|
||||
return str(value)
|
||||
|
||||
# 当isLabeled列为1时,整行前景变色
|
||||
if role == Qt.ForegroundRole:
|
||||
if "isLabeled" in self._dataframe.columns:
|
||||
try:
|
||||
print(self._showdata.iloc[row]["Index"])
|
||||
if role == Qt.ForegroundRole and "isLabeled" in self._dataframe.columns:
|
||||
try:
|
||||
print(self._showdata.iloc[row]["Index"])
|
||||
|
||||
event_index = self._showdata.iloc[row]["Index"]
|
||||
is_labeled_value = self._dataframe[self._dataframe["Index"] == event_index]["isLabeled"]
|
||||
if is_labeled_value.empty:
|
||||
return None
|
||||
is_labeled_value = is_labeled_value.values[0]
|
||||
|
||||
if is_labeled_value == 1:
|
||||
# remark 返回蓝色 score 1 返回绿色, 2返回橙色 3返回灰色 -1返回红色
|
||||
# 获取此行的remark
|
||||
remark = not self._dataframe[self._dataframe["Index"] == event_index]["remark"].values[0]
|
||||
score = self._dataframe[self._dataframe["Index"] == event_index]["score"].values[0]
|
||||
if remark == "待讨论":
|
||||
return QColor(0, 0, 255)
|
||||
elif score == 1:
|
||||
return QColor(0, 255, 0)
|
||||
elif score == 2:
|
||||
return QColor(255, 165, 0)
|
||||
elif score == 3:
|
||||
return QColor(128, 128, 128)
|
||||
elif score == -1:
|
||||
return QColor(255, 0, 0)
|
||||
else:
|
||||
# 返回黑色
|
||||
return QColor(0, 0, 0)
|
||||
|
||||
else:
|
||||
# 返回灰色
|
||||
return QColor(0, 0, 0)
|
||||
except Exception as e:
|
||||
event_index = self._showdata.iloc[row]["Index"]
|
||||
is_labeled_value = self._dataframe[self._dataframe["Index"] == event_index]["isLabeled"]
|
||||
if is_labeled_value.empty:
|
||||
return None
|
||||
is_labeled_value = is_labeled_value.values[0]
|
||||
|
||||
if is_labeled_value != 1:
|
||||
return QColor(0, 0, 0)
|
||||
|
||||
# remark 返回蓝色 score 1 返回绿色, 2返回橙色 3返回灰色 -1返回红色
|
||||
# 获取此行的remark
|
||||
remark = not self._dataframe[self._dataframe["Index"] == event_index]["remark"].values[0]
|
||||
score = self._dataframe[self._dataframe["Index"] == event_index]["score"].values[0]
|
||||
if remark == "待讨论":
|
||||
return QColor(0, 0, 255)
|
||||
elif score == 1:
|
||||
return QColor(0, 255, 0)
|
||||
elif score == 2:
|
||||
return QColor(255, 165, 0)
|
||||
elif score == 3:
|
||||
return QColor(128, 128, 128)
|
||||
elif score == -1:
|
||||
return QColor(255, 0, 0)
|
||||
else:
|
||||
# 返回黑色
|
||||
return QColor(0, 0, 0)
|
||||
|
||||
except Exception as e:
|
||||
return None
|
||||
|
||||
return None
|
||||
|
||||
|
||||
Reference in New Issue
Block a user