能否将非零元素标记为红色
时间: 2024-04-16 22:28:51 浏览: 14
你可以使用 pandas 库和 openpyxl 库来处理 .csv 文件,并将非零元素标记为红色。下面是一个示例代码,它将非零元素标记为红色,并将结果保存为一个新的 .xlsx 文件:
```python
import pandas as pd
from openpyxl.styles import PatternFill
# 读取原始 .csv 文件
df = pd.read_csv('your_file.csv', header=None)
# 创建一个新的 Excel 文件
writer = pd.ExcelWriter('result.xlsx', engine='openpyxl')
# 将数据写入 Excel 文件的第一个 sheet
df.to_excel(writer, sheet_name='Sheet1', index=False, header=False)
# 打开 Excel 文件并获取第一个 sheet
book = writer.book
sheet = book['Sheet1']
# 遍历每个单元格,将非零元素标记为红色
red_fill = PatternFill(start_color='FFFF0000', end_color='FFFF0000', fill_type='solid') # 红色填充
for row in sheet.iter_rows(min_row=1, min_col=1, max_row=df.shape[0], max_col=df.shape[1]):
for cell in row:
if cell.value != 0:
cell.fill = red_fill
# 保存 Excel 文件
writer.save()
```
请确保将 `'your_file.csv'` 替换为你要处理的实际 .csv 文件的文件名。运行这段代码后,你将得到一个新的 `result.xlsx` 文件,其中非零元素被标记为红色。