蜂群图“渐变”

pip install matplotlib seaborn

import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np
import pandas as pd

# 生成示例数据
np.random.seed(42)
data = pd.DataFrame({
    'Category': np.random.choice(['A', 'B', 'C'], size=300),
    'Value': np.random.randn(300)
})

# 创建颜色渐变
cmap = plt.get_cmap('viridis')
norm = plt.Normalize(data['Value'].min(), data['Value'].max())
colors = cmap(norm(data['Value']))

# 创建蜂群图
plt.figure(figsize=(10, 6))
ax = sns.swarmplot(x='Category', y='Value', data=data, palette=colors)

# 添加颜色条
sm = plt.cm.ScalarMappable(cmap=cmap, norm=norm)
sm.set_array([])
plt.colorbar(sm, ax=ax, label='Value')

plt.title('Bee Swarm Plot with Gradient Colors')
plt.show()