我有一个文件,其中有绿色突出显示的单元格和红色突出显示的单元格,表示严重性级别
现在我正在使用StyleFrame来提取它们,但是由于某些原因,我可以提取红细胞,但不能提取绿细胞。你知道怎么做吗
注意,我确保代码中指定的颜色与文件中的颜色相同
from StyleFrame import StyleFrame, utils
GeneralCategories = StyleFrame.read_excel('Categories.xlsx', read_style=True, use_openpyxl_styles=False)
def redBackground(cell):
return cell if cell.style.bg_color in {utils.colors.red, 'FFFF0000'} else np.nan
def greenBackground(cell):
return cell if cell.style.bg_color in {utils.colors.green, '00FF00'} else np.nan
SevereCategory = StyleFrame(GeneralCategories.applymap(redBackground).dropna(axis=(0, 1), how='all'))
NonSevereCategory = StyleFrame(GeneralCategories.applymap(greenBackground).dropna(axis=(0, 1), how='all'))
结果:
print(SevereCategory)
Keyword Categories
1 Adult Content: Nudity & Partial Nudity
2 Adult Content: Nudity & Partial Nudity;Adult C...
3 Adult Content: Nudity & Partial Nudity;Adult C...
4 Adult Content: Nudity & Partial Nudity;Adult C...
5 Adult Content: Nudity & Partial Nudity;Adult C...
.. ...
print(NonSevereCategory)
Empty DataFrame
Columns: []
Index: []
感谢@OsmosisJonesLoL,我在通常的6位绿色十六进制代码之前添加了“FF”,它成功了
最终代码:
相关问题 更多 >
编程相关推荐