dataframe添加背景色并保存
时间: 2023-07-11 20:26:00 浏览: 48
要为 DataFrame 添加背景色,可以使用 pandas.style 库。以下是一个示例代码:
```python
import pandas as pd
# 创建一个 DataFrame
data = {'name': ['Alice', 'Bob', 'Charlie', 'David'],
'age': [25, 30, 35, 40],
'gender': ['F', 'M', 'M', 'M']}
df = pd.DataFrame(data)
# 创建一个样式函数,为 age 列大于 30 的行添加背景色
def highlight_age(row):
if row['age'] > 30:
return ['background-color: yellow'] * len(row)
else:
return [''] * len(row)
# 应用样式函数到 DataFrame
styled_df = df.style.apply(highlight_age, axis=1)
# 将带有样式的 DataFrame 保存为 HTML 文件
styled_df.to_excel('styled_df.xlsx', engine='openpyxl', index=False)
```
在这个例子中,我们定义了一个名为 highlight_age 的函数,该函数用于为 age 列大于 30 的行添加黄色背景色。然后,我们将这个样式函数应用到 DataFrame 上,并将结果保存为一个 Excel 文件。