如何从 Bokeh 2.4.3 中的 ColumnDataSource 中提取列名？

Question

我正在使用函数创建图表，我想使用来自 DF 的列名称，并将它们用作图表中的标题。

如果我有以下情况：

from bokeh.plotting import figure
from bokeh.models import ColumnDataSource
import pandas as pd
import numpy as np      
        
df = pd.DataFrame(np.random.randint(0,100,size=(10, 2)), columns=('Cats', 'Dogs'))

source = ColumnDataSource(df)

然后我怎样才能得到列名，我在想使用

source.data

但无法弄清楚究竟如何..

Answer 1

如果你打印

source

，你会看到它是一个带有

data

的对象。

数据是字典，键是列名，

'index'

是 DataFrame 的索引。

print(source.data)
>>> {
  'index': array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]),
  'Cats': array([87, 56, 98, 10, 30, 89, 40, 51, 17, 13]),
  'Dogs': array([ 9, 11, 73, 76, 67, 46, 64, 62, 93, 71])
}

和

print(source.data.keys())
>>> dict_keys(['index', 'Cats', 'Dogs'])