来自多个数据框列的分类数据的散点图

Question

我有一个这样的 DataFrame df：

   name         Int1        Int2         Int3
0     foob -2.534519e-05 -1.156744e-04 -1.410195e-04
1     arz  2.907239e-04  3.502863e-04  6.410102e-04
2     foo2 -2.140769e-04  4.214626e-04  2.073857e-04
3     bar  3.366116e-03 -6.125303e-04  2.753586e-03
4     rnd -5.014413e-05 -6.740579e-06 -5.688471e-05
5     baz  3.334906e-04 -7.846232e-05  2.550283e-04
6     rnd2 -6.111069e-04  2.194443e-03  1.583336e-03
7     tet  3.184057e-04  2.208398e-04  5.392455e-04

df 应绘制为散点图，描述每个名称的三个数据点（Int1、Int2、Int3）。

目前，我正在使用 seaborn 的 stripplot 函数，如果我分配每一列的图（例如 x=name, y= int1) 到一个图的同轴:

fig, ax = plt.subplots()
seaborn.stripplot(df.name, df.Int1, ax=ax, c='red')
seaborn.stripplot(df.name, df.Int2, ax=ax, c='blue')

但是，我想以更好的方式绘制它，主要原因是拥有适当的图例和更好的定制。解决方案也可以基于pandas

Answer 1

这是我的解决方案。其实很简单:

df_melt=pd.melt(df,id_vars=['name'], var_name='intensities', value_name='values')
sns.stripplot(x="name", y="values", data=df_melt, hue='intensities')

这采用原始 df 并在 pandas melt 函数的帮助下，生成一个包含单个 intensities 列，每个名称的 Int1、Int2 和 Int3 值各占一行。第二行使用 seaborn 的 stripplot 来绘制 df_melt 并根据相应的列 Int1、Int2 和 [=16] 着色=]

来自多个数据框列的分类数据的散点图

Scatter plot of categorical data from multiple data frame columns

python

matplotlib

dataframe

pandas

seaborn