通过传递列表来索引字典

Question

我正在尝试使用存储在字典中的颜色来绘制数据框。 此代码有效：

import matplotlib.pyplot as plt
import pandas as pd

df = pd.DataFrame(np.random.randn(100, 4), index=range(100), columns=list('ABCD'))
df = df + [-2, 0, 2, 4]
x = list(range(100, 1100, 10))
clrhex = {'abcd':'#FFA500', 'efgh':'#FF0000','ijkl':'#000000','mnop':'#6495ED'}

plt.plot(x, df, color=[clrhex['abcd'], clrhex['efgh'], clrhex['ijkl'], clrhex['mnop']])
plt.show()

但我想通过使用列表索引字典来缩短它。这个：

plt.plot(x, df, color=clrhex['abcd', 'efgh', 'ijkl', 'mnop'])

给出以下错误信息：

Invalid RGBA argument

如何从字典中获取多个值？

Answer 1

如果你使用 matplotlib > 1.5，使用set_color_cycle 会弹出一条折旧消息：

c:\Python27\lib\site-packages\matplotlib\cbook\deprecation.py:106:
MatplotlibDeprecationWarning：set_color_cycle 属性为 在 1.5 版中已弃用。请改用 set_prop_cycle。

所以，使用set_prop_cycle 你可以这样做：

df = pd.DataFrame(np.random.randn(100, 4), index=range(100), columns=list('ABCD'))
df = df + [-2, 0, 2, 4]
x = list(range(100, 1100, 10))
clrhex = {'abcd':'#FFA500', 'efgh':'#FF0000','ijkl':'#000000','mnop':'#6495ED'}

plt.gca().set_prop_cycle('color', [clrhex[i] for i in ['abcd', 'efgh', 'ijkl', 'mnop']])
plt.plot(x, df)

plt.show()

使用set_prop_cycle 的优点是您可以将它与其他东西一起使用，而不仅仅是颜色，即线条样式或线宽。下面是一些修改后的代码，展示了如何更改颜色和线宽（如果您愿意的话），并使用面向对象的 API：

from cycler import cycler

df = pd.DataFrame(np.random.randn(100, 4), index=range(100), columns=list('ABCD'))
df = df + [-2, 0, 2, 4]
x = list(range(100, 1100, 10))
clrhex = {'abcd':'#FFA500', 'efgh':'#FF0000','ijkl':'#000000','mnop':'#6495ED'}

fig, ax = plt.subplots()

ax.set_prop_cycle(cycler('color', [clrhex[i] for i in ['abcd', 'efgh', 'ijkl', 'mnop']]) + 
    cycler('lw', [1,2,3,4]))

ax.plot(x, df)

plt.show()

这给出了：

Answer 2

嗯，你可以用更少的字符来理解：

color=[clrhex[i] for i in ['abcd', 'efgh', 'ijkl', 'mnop']]

编辑

你的代码对我不起作用，所以我使用了这个：

plt.plot(x, df)
plt.gca().set_color_cycle([clrhex[i] for i in ['abcd', 'efgh', 'ijkl', 'mnop']])
plt.show()