Python Bokeh:根据columndatasource中的列设置线条颜色

时间:2015-09-18 06:29:10

标签: python pandas bokeh

我正在尝试生成一个包含多行的图表,但我使用的数据通常是这样的长形式:

x = [0.1, 0.5, 1.0, 1.5, 2.0, 2.5, 3.0]
y0 = [i**2 for i in x]
y1 = [10**i for i in x]
y2 = [10**(i**2) for i in x]
df = pandas.DataFrame(data=[x,y0,y1,y2]).T
df.columns = ['x','y0','y1','y2']

df2 = pd.concat([df.iloc[0:,1],df.iloc[0:,2],df.iloc[0:,3]], axis=0, keys = ['a','b','c']).reset_index()
df2.columns = ['grp','x','y']

df2

+----+-----+---+----------+
|    | grp | x |    y     |
+----+-----+---+----------+
|  0 | a   | 0 | 0.01     |
|  1 | a   | 1 | 0.25     |
|  2 | a   | 2 | 1.00     |
|  3 | a   | 3 | 2.25     |
|  4 | a   | 4 | 4.00     |
|  5 | a   | 5 | 6.25     |
|  6 | a   | 6 | 9.00     |
|  7 | b   | 0 | 1.26     |
|  8 | b   | 1 | 3.16     |
|  9 | b   | 2 | 10.00    |
| 10 | b   | 3 | 31.62    |
| 11 | b   | 4 | 100.00   |
| 12 | b   | 5 | 316.23   |
| 13 | b   | 6 | 1,000.00 |
+----+-----+---+----------+

cd_df2 = ColumnDataSource(df2)

也就是说,我将拥有'群组',其中每个群组的x,y对在多行中列出。

以下产生所有3行,但它们都显示为灰色。设置color ='grp'并不特定于columndata源中grp字段中每个值的颜色

f = figure(tools="save",y_axis_type="log", y_range=[0.001, 10**11], 
       title="log axis example",x_axis_label='sections', y_axis_label=
       'particles')

f.line('x','y', line_color = 'grp', source = cd_df2)

我怎样才能在bokeh.plotting或bokeh.models api中实现这一点(想要避免使用高级图表来更好地理解库)?我愿意接受其他建议,避免为每一行显式调用f.line()一次并单独设置颜色(我可能有10行以上,这会很繁琐)。

1 个答案:

答案 0 :(得分:1)

您可以重复使用this example中描述的原则。它基于“补丁”,但对于“行”是相同的(见http://bokeh.pydata.org/en/latest/docs/reference/plotting.html):

p = figure()    
p.patches([x2 for a in areas], list(areas.values()), color=colors, alpha=0.8, line_color=None)