试图在seaborn中使用3种不同的向量进行模拟。我想让每个向量都具有不同的x坐标和不同的颜色。
不幸的是,所有教程都以某种格式提供数据,我无法真正找到其解释/手册……这是我到目前为止所掌握的:
#!/usr/bin/env python3
import numpy as np
import matplotlib.pyplot as plt
import seaborn
import pandas as pd
# Create figure
fig = plt.figure()
ax = fig.add_subplot(111)
ax.set_ylabel('Evaluations')
colors = [
'tab:orange',
'tab:green',
'tab:red',
]
labels = [
'Method 2',
'Method 3',
'Method 4',
]
data = [
[1, 2.1, 3.2, 4.5, 3.6, 2.7, 1.4],
[2.2, 4.7, 5.1, 4.4, 3.8, 5, 3.4],
[8.4, 7.2, 6.1, 5.4, 8.1, 7.4, 6.8],
]
data = np.array(data).T
df = pd.DataFrame(data, columns=labels)
seaborn.swarmplot(data=df, y='Method 2', color=colors[0])
seaborn.swarmplot(data=df, y='Method 3', color=colors[1])
seaborn.swarmplot(data=df, y='Method 4', color=colors[2])
plt.show()
这几乎可以工作,但是将所有内容绘制在同一轴上:
此外,标签应位于x轴而不是y轴。可以肯定的是,我在这里确实缺少一些基本知识。有人吗?
答案 0 :(得分:1)
尝试:
df2 = df.melt()
colors = ["orange", "green", "red"]
sns.swarmplot(data=df2, x = 'variable', y='value', palette =colors)
plt.show()