我正在使用seaborn和pandas從不同(但相關)的數據中創建一些條形圖.這兩個數據集共享一個用作色調的公共類別,因此我想確保在這兩個圖中該類別的條形顏色匹配.我怎么能這樣做？

一個基本的例子如下：

import seaborn as sns

import pandas as pd

import matplotlib.pyplot as plt

sns.set_style('darkgrid')

fig, ax = plt.subplots()

a = pd.DataFrame({'Program': ['A', 'A', 'B', 'B', 'Total', 'Total'],

'Scenario': ['X', 'Y', 'X', 'Y', 'X', 'Y'],

'Duration': [4, 3, 5, 4, 9, 7]})

g = sns.barplot(data=a, x='Scenario', y='Duration',

hue='Program', ci=None)

plt.tight_layout()

plt.savefig('3 progs.png')

plt.clf()

b = pd.DataFrame({'Program': ['A', 'A', 'B', 'B', 'C', 'C', 'Total', 'Total'],

'Scenario': ['X', 'Y', 'X', 'Y', 'X', 'Y', 'X', 'Y'],

'Duration': [4, 3, 5, 4, 3, 2, 12, 9]})

g = sns.barplot(data=b, x='Scenario', y='Duration',

hue='Program', ci=None)

plt.tight_layout()

plt.savefig('4 progs.png')

制作兩張圖：

在這個例子中,我想確保Total類別在兩個圖形中使用相同的顏色(例如黑色)

A.使用顏色列表

確保兩個圖中相同類別具有相同顏色的最簡單解決方案是在創建繪圖時手動指定顏色.

# First bar plot

ax = sns.barplot(data=a, x='Scenario', y='Duration',

hue='Program', ci=None, palette=["C0", "C1", "k"])

# ...

# Second bar plot

ax2 = sns.barplot(data=b, x='Scenario', y='Duration',

hue='Program', ci=None, palette=["C0", "C1","C2", "k"])

顏色“C2”(顏色循環的第三種顏色)僅存在于存在程序C的第二個圖中.

B.使用字典

您也可以使用字典,將色調列中的值映射到顏色,而不是列表.

palette ={"A":"C0","B":"C1","C":"C2", "Total":"k"}

ax = sns.barplot(data=a, x='Scenario', y='Duration', hue='Program', palette=palette)

# ...

ax2 = sns.barplot(data=b, x='Scenario', y='Duration', hue='Program', palette=palette)

在這兩種情況下,輸出將如下所示：

C.自動字典

最后,您可以根據色調列中的值自動創建此字典.這里的優點是你既不需要知道顏色,也不需要知道各個數據幀之前的值.

import seaborn as sns

import pandas as pd

import matplotlib.pyplot as plt

sns.set_style('darkgrid')

fig, ax = plt.subplots()

a = pd.DataFrame({'Program': ['A', 'A', 'B', 'B', 'Total', 'Total'],

'Scenario': ['X', 'Y', 'X', 'Y', 'X', 'Y'],

'Duration': [4, 3, 5, 4, 9, 7]})

b = pd.DataFrame({'Program': ['A', 'A', 'B', 'B', 'C', 'C', 'Total', 'Total'],

'Scenario': ['X', 'Y', 'X', 'Y', 'X', 'Y', 'X', 'Y'],

'Duration': [4, 3, 5, 4, 3, 2, 12, 9]})

unique = a["Program"].append(b["Program"]).unique()

palette = dict(zip(unique, sns.color_palette()))

palette.update({"Total":"k"})

ax = sns.barplot(data=a, x='Scenario', y='Duration',

hue='Program', ci=None, palette=palette)

plt.tight_layout()

plt.figure()

ax2 = sns.barplot(data=b, x='Scenario', y='Duration',

hue='Program', ci=None, palette=palette)

plt.tight_layout()

plt.show()

總結

以上是生活随笔為你收集整理的python颜色列表代码seaborn_在Python中Seaborn – 根据色调名称更改条形颜色的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。