pandas 设置表格的只把表头行填充为浅蓝色,全表加边框,并输出到EXCEL,代码要简单
时间: 2024-05-01 20:19:39 浏览: 265
你可以使用 pandas 库中的 Styler 对象来实现这个需求,代码如下:
``` python
import pandas as pd
# 创建一个 DataFrame
df = pd.DataFrame({'Name': ['Alice', 'Bob', 'Charlie'],
'Age': [25, 30, 35],
'Gender': ['Female', 'Male', 'Male']})
# 定义函数,用于将表头行的背景色设为浅蓝色
def highlight_columns(s):
return ['background-color: lightblue' if i == 0 else '' for i in range(len(s))]
# 创建 Styler 对象,并设置样式
styler = df.style.apply(highlight_columns, axis=1)\
.set_properties(**{'border': '1px solid black', 'text-align': 'center'})\
.set_table_styles([{'selector': 'thead', 'props': [('font-weight', 'bold')]}])\
.set_caption('My table')
# 输出到 Excel
styler.to_excel('output.xlsx', engine='openpyxl', index=False)
```
这段代码中,首先创建了一个 DataFrame,然后定义了一个函数 `highlight_columns`,该函数用于将表头行的背景色设置为浅蓝色。然后,创建一个 Styler 对象,并调用 `apply` 方法,在它的基础上通过 `set_properties` 和 `set_table_styles` 方法设置样式。最后,通过 `to_excel` 方法将样式应用到表格中,并输出到 Excel 文件。
阅读全文