我有这样的事情:
import holoviews as hv
import pandas as pd
from holoviews import opts, dim
hv.extension('bokeh')
renderer = hv.renderer('bokeh')
csv_path = r'C:\Users\jose\Downloads\enron-v1.csv'
df_csv = pd.read_csv(csv_path ,index_col=0)
df_filter = df_csv[["fromJobtitle", "toJobtitle"]]
df_final = df_filter.groupby(df_filter.columns.tolist(),as_index=False).size()
这会产生这样的结果:
fromJobtitle toJobtitle size
0 CEO CEO 65
1 CEO Director 23
2 CEO Employee 56
3 CEO In House Lawyer 7
4 CEO Manager 104
我想提取这个矩阵作为在holoviews中将数据绘制为sankey图的一种方式:
[['CEO', 'CEO', 65],
['CEO', 'Director', 23],
['CEO', 'Employee', 56]]
.......etc
答案 0 :(得分:0)
pd.DataFrame
已经以这种格式存储它们,所以你只需要这样做:
df_final.values
Out[149]:
array([['CEO', 'CEO', 65],
['CEO', 'Director', 23],
['CEO', 'Employee', 56],
['CEO', 'Lawyer', 7],
['CEO', 'Manager', 104]], dtype=object)