使用python为列文件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>；一,

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')

我不知道如何使用熊猫的风格，你能帮我这个案子吗

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

#编辑：您可以将样式化的数据框导出到#excel

#edit2：将列“Cod”的de颜色改为列“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')