根据数据框中 Opportunity 的递增值，为从红色到绿色的颜色代码添加一个新列

Question

我有一个数据框，我想为颜色代码生成一个新的列，它从红色开始代表机会的最低价值，然后向绿色移动以获得机会的最高价值

我的数据框 -

State       Brand       DYA  Opportunity    

Jharkhand   Ariel     0.15   0.00853    
Jharkhand   Fusion    0.02   0.00002
Jharkhand   Gillett   0.04   -0.0002

Answer 1

要获得从红色到绿色的颜色范围，您可以使用 matplotlib color maps ，更具体地说，是RdYlGn 。 但在应用颜色映射之前，首先，您需要将Opportunity列中的数据标准化为 0 和 1 之间的值。然后，您可以从这里以您认为合适的任何方式将数据编码为颜色代码。 例如，我在这里使用 Pandas apply function 和rbg2hex ，目的是获取表示Opportunity列中使用的颜色映射的 CSS 值。

import matplotlib.cm as cm
from matplotlib.colors import Normalize
from matplotlib.colors import rgb2hex

# df = Your original dataframe

cmapR = cm.get_cmap('RdYlGn')
norm = Normalize(vmin=df['Opportunity'].min(), vmax=df['Opportunity'].max())

df['Color'] = df['Opportunity'].apply(lambda r: rgb2hex(cmapR(norm(r))))

dstyle = df.style.background_gradient(cmap=cmapR, subset=['Opportunity'])
dstyle.to_html('sample.html')