如何根据数据框（pandas）中的值设置条形（matplotlib）的颜色？

Question

我想从数据框创建条形图。 但我想根据数据框中“红色”列中的值为每个条着色。 我有以下代码：

plt.bar(df.index, df['Mean'], yerr = df['yerr'], capsize=7, color = (df['red'], 0, 0, 0.6))

我想从列“ red”（从0到1）中取值，但是它一直失败。 你会怎么做？

Answer 1

这样的事情会起作用。 您必须创建具有共同长度的蓝色，绿色和alpha列，然后将它们与红色一起压缩。

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

data = [
    [1, 1, 0.1],
    [2, 1, 0.2],
    [3, 1, 0.3],
    [4, 1, 0.4],
    [5, 1, 0.5],
    [6, 1, 0.6],
    [7, 1, 0.7],
    [8, 1, 0.8],
    [9, 1, 0.9],
]
columns = ['Mean', 'yerr', 'red']
df = pd.DataFrame(data=data, columns=columns)

r = df['red']
g = np.zeros(r.shape[0])
b = np.zeros(r.shape[0])
a = np.ones(r.shape[0]) * 0.6
plt.bar(df.index, df['Mean'], yerr=df['yerr'], capsize=7,
        color=list(zip(r, g, b, a)))

plt.show()