您好：程式码如下：想请问一下，该如何有LABLE对照表，与颜色对应比如 台积电 blue鸿海 green 颜色显示于LABLE

谢谢

from bokeh.plotting import figure, output_file, show
from bokeh.plotting import ColumnDataSource
from bokeh.models import CategoricalColorMapper, Legend, LegendItem
import pandas as pd

# 汇入CSV格式的档案
df = pd.read_csv("tech_stocks_2017.csv", encoding="utf8")

output_file("Ch16_4_3d.html")

tech_stocks = ["台积电", "鸿海", "广达", "联发科", "和硕"]

c_map = CategoricalColorMapper(
factors=tech_stocks,
palette=["blue","green","red","yellow","gray"])

data = ColumnDataSource(data={
"close": df["Close"],
"volume": df["Volume"],
"name": df["Name"]
})

p = figure(title="苹概科技股的收盘价与成交量",
height=400, width=700,
x_range=(min(df.Close), max(df.Close)),
y_range=(min(df.Volume), max(df.Volume))
)

p.diamond(x="close", y="volume", source=data,
color={"field": "name", "transform": c_map } )

p.xaxis.axis_label = "2017年收盘价"
p.yaxis.axis_label = "2017年成交量"

show(p)

1 个回答

1

zivzhong

iT邦研究生 4 级 ‧ 2025-01-24 01:50:41

最佳解答

试试:

from bokeh.plotting import figure, output_file, show
from bokeh.plotting import ColumnDataSource
from bokeh.models import CategoricalColorMapper, Legend, LegendItem
import pandas as pd

# 汇入CSV格式的档案
df = pd.read_csv("tech_stocks_2017.csv", encoding="utf8")

output_file("Ch16_4_3d.html")

tech_stocks = ["台积电", "鸿海", "广达", "联发科", "和硕"]

c_map = CategoricalColorMapper(
           factors=tech_stocks, 
           palette=["blue","green","red","yellow","gray"])

data = ColumnDataSource(data={
        "close": df["Close"],
        "volume": df["Volume"],
        "name": df["Name"]
        })

p = figure(title="苹概科技股的收盘价与成交量", 
           height=400, width=700, 
           x_range=(min(df.Close), max(df.Close)),
           y_range=(min(df.Volume), max(df.Volume)),
           tools="pan,wheel_zoom,box_zoom,reset,save")

# 添加 diamond glyph，并使用 legend_field 自动生成 Legend
p.diamond(x="close", y="volume", source=data,
          color={"field": "name", "transform": c_map},
          legend_field="name",  # 设置 legend_field
          size=10, alpha=0.6)

p.xaxis.axis_label = "2017年收盘价"
p.yaxis.axis_label = "2017年成交量"

# 设置 Legend 的位置和样式
p.legend.title = "科技股"
p.legend.location = "top_left"
p.legend.background_fill_alpha = 0.6

show(p)


  • 1

noway

iT邦研究生 1 级 ‧
2025-01-24 18:50:01

谢谢您的帮忙

修改