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

一個(gè)基本的例子如下：

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

制作兩張圖：

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

A.使用顏色列表

確保兩個(gè)圖中相同類別具有相同顏色的最簡單解決方案是在創(chuàng)建繪圖時(shí)手動(dòng)指定顏色.

# 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”(顏色循環(huán)的第三種顏色)僅存在于存在程序C的第二個(gè)圖中.

B.使用字典

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

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.自動(dòng)字典

最后,您可以根據(jù)色調(diào)列中的值自動(dòng)創(chuàng)建此字典.這里的優(yōu)點(diǎn)是你既不需要知道顏色,也不需要知道各個(gè)數(shù)據(jù)幀之前的值.

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

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。