尝试将颜色渐变添加到Matplotlib图表

时间:2018-06-08 21:21:53

标签: python matplotlib

我正在尝试为我的图表添加红色到绿色渐变。然而,当我运行以下内容时,我得到:

TypeError: object of type 'Color' has no len()

以下是相关代码:

from colour import Color

red = Color("red")
colors = list(red.range_to(Color("green"),10))

for col in ['DISTINCT_COUNT', 'NULL_COUNT','MAX_COL_LENGTH', 'MIN_COL_LENGTH']: 
    grid[['COLUMN_NM', col]].set_index('COLUMN_NM').plot.bar(title=table_nm, figsize=(12, 8), color=colors)
    plt.xlabel('Column', labelpad=12)
    plt.tight_layout()
    plt.show()

如果我只是运行顶部并打印结果,它似乎运行良好:

red = Color("red")
colors = list(red.range_to(Color("green"),10))
print(colors)

[<Color red>, <Color #f13600>, <Color #e36500>, <Color #d58e00>, <Color #c7b000>, <Color #a4b800>, <Color #72aa00>, <Color #459c00>, <Color #208e00>, <Color green>]

所以我必须在这里尝试使用它:

grid[['COLUMN_NM', col]].set_index('COLUMN_NM').plot.bar(title=table_nm, figsize=(12, 8), color=colors)

有什么想法吗?

1 个答案:

答案 0 :(得分:1)

Matplotlib无法与colour.Color个实例一起使用。如果您愿意,可以将它们转换为RGB值。

接下来,熊猫不喜欢给出几种颜色。但是你可以改用matplotlib图。

import matplotlib.pyplot as plt
import pandas as pd
from colour import Color

df = pd.DataFrame({"x" : list(range(3,13))})

red = Color("red")
colors = list(red.range_to(Color("green"),10))
colors = [color.rgb for color in colors]

plt.bar(df.index, df["x"], color=colors)
plt.xlabel('Column', labelpad=12)
plt.tight_layout()
plt.show()

enter image description here

请注意,通常您更愿意使用色彩映射。您可以使用标准化值来调用此色彩图,您可以根据这些值对条形图进行着色。

import matplotlib.pyplot as plt
import matplotlib.colors as mcolors
import pandas as pd
import numpy as np

df = pd.DataFrame({"x" : np.random.rand(10)*10})

cmap = mcolors.LinearSegmentedColormap.from_list("", ["red", "yellow", "green"])

plt.bar(df.index, df["x"], color=cmap(df.x.values/df.x.values.max()))
plt.xlabel('Column', labelpad=12)
plt.tight_layout()
plt.show()

enter image description here