在seaborn barplot中按列值定义色调

Question

我希望列的颜色由它们在 x 轴上的值确定，例如，在 x 轴上具有相同值的条应该分配给它们相同的颜色。

import seaborn as sns
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd 

df = pd.DataFrame(index=['A','B','C','D','E','F'],data={'col1':np.array([2.3423,4.435,9.234,9.234,2.456,6.435])})
ax = sns.barplot(x='col1', y=df.index.values, data=df,palette='magma')

这是目前使用默认设置的样子。我认为有一种简单优雅的方法可以做到这一点，但对任何解决方案都感兴趣。

Answer 1

这里有一个解决方案：

import seaborn as sns
import matplotlib as mpl, matplotlib.pyplot as plt
import numpy as np
import pandas as pd

df = pd.DataFrame(index=['A','B','C','D','E','F'],
  data={'col1':np.array([2.3423,4.435,9.234,9.234,2.456,6.435])})
ax = sns.barplot(x='col1', y=df.index.values, data=df,
  palette=mpl.cm.magma(df['col1']*.1))

注意：mpl.cm.magma是一个Colormap实例，用于将数据值（浮点数）从区间 [0, 1] 转换为Colormap代表的颜色。如果您想要“自动缩放”您的数据值，您可以palette=mpl.cm.ScalarMappable(cmap='magma').to_rgba(df['col1'])在sns.barplot()调用中使用。

这里的输出：