如何使用Bokeh和Pandas绘制带有分类数据的散点图?

时间:2017-02-14 02:02:51

标签: python pandas bokeh

我有一个包含数据的Pandas DataFrame:

Merchant | Non-Merchant | Num of Chats Received | Num of Chats Made
1        | 0            | 80                    | 50
0        | 1            | 60                    | 30
1        | 0            | 70                    | 40
0        | 1            | 50                    | 20

我希望能够在散点图上绘制两种不同类型的用户(商家,非商家),比较[收到的聊天数,y轴]。

在上面的数据中,

商家是那些被标记为' 1'在商家专栏和

非商家是标记为' 1'在商家专栏中,

它是二进制的,没有[1,1]。

我是Bokeh和Data Viz的新手,假设Bokeh是我的首选方法,我应该怎么做?

1 个答案:

答案 0 :(得分:1)

这是使用散景

的一种方法

CODE:

from bokeh.plotting import figure, output_file, show
import pandas as pd

data = {'Merchant':[1,0,1,0],
        'Non-Merchant':[0,1,0,1],
        'Num of Chats Received':[80,60,70,50],
        'Num of Chats Made':[50,30,40,20]}

df = pd.DataFrame(data)

merchant_chats_made = list(df[df['Merchant'] == 1]['Num of Chats Made'])
merchant_chats_received = list(df[df['Merchant'] == 1]['Num of Chats Received'])
non_merchant_chats_made = list(df[df['Non-Merchant'] == 1]['Num of Chats Made'])
non_merchant_chats_received = list(df[df['Non-Merchant'] == 1]['Num of Chats Received'])

output_file('merchant.html')

p = figure(plot_width=400, plot_height=400)
p.circle(merchant_chats_made,
         merchant_chats_received,
         size=10,color='red',alpha=0.5,
         legend='Merchant')

p.circle(non_merchant_chats_made,
         non_merchant_chats_received,
         size=10,color='blue',alpha=0.5,
         legend='Non-Merchant')

p.xaxis.axis_label = 'Chats Made'
p.yaxis.axis_label = 'Chats Received'
p.legend.location = 'top_left'

show(p)

enter image description here