python - 用 python 为 pandas 的列文件 csv 添加颜色

Question

我正在尝试读取数据文件 csv，已完成，但我想在某些条件下写入另一个数据，为 csv 文件中的数据添加样式(颜色，两者...)。

i) 我从文件 csv 中读取数据，先是一个简单的测试文件，然后再应用一个大数据文件，所以数据测试文件是

cod     price   var
AMZ     8484    -1.15
BABA    1390    3.5
GOGL    327     -10.5
MSFT    1135    3.3
VISA    6499    2.0
PBB     1055    4.0
CHC     3455    1.1
IFT     5228    -4.0
FRA     4555    2.0

ii) 将条件和只读值放入列 var > 1

iii) 在 cod 列中放回底色:

 red where var => 3 
 yellow where var = 2 <3

iv) 保存在另一个数据csv中

我的代码是:

import pandas as pd

df = pd.read_csv("teste.csv") 
f1 = df['var'] > 1    


df[f1].to_csv('export.csv')

我不知道如何在 pandas 中使用样式，你能帮我解决这个问题吗？

Answer 1

你可以试试这样的东西

import numpy as np
import pandas as pd


df = pd.DataFrame([[8484,-1.15],[1390,3.5],[327,10.5],[1135,3.3],[6499,2.0],[1055,4.0]], columns=['data1','dat2'])


def color_rule(val):
    return ['background-color: red' if x >= 3 else 'background-color: yellow' for x in val]

html_column = df.style.apply(color_rule, axis=1, subset=['dat2'])

html_column.to_excel('styled.xlsx', engine='openpyxl')

html_column

#edit:您可以使用 to_excel 导出样式化数据框

#edit2:更改“Cod”列的颜色而不是“var”列

import numpy as np
import pandas as pd


df = pd.DataFrame([['ex1',8484,-1.15],['ex2',1390,3.5],['ex3',327,10.5],['ex4',1135,3.3],['ex5',6499,2.0],['ex6',1055,4.0]], columns=['tag','data1','dat2'])


def color_rule(tag):
    var1 = df['dat2'][df['tag'] == tag.values[0]]
    return ['background-color: red' if x >= 3 else 'background-color: yellow' for x in var1]

html_column = df.style.apply(color_rule, axis=1, subset=['tag'])

html_column.to_excel('styled.xlsx', engine='openpyxl')