python Seaborn - 制作线图的子图

Posted 2021-05-09

# Plot results
f, ax = plt.subplots(2, 1, figsize=(10, 12), sharex=True)
sns.despine()
for i, question_condition in enumerate(main_results["question_conditions"].unique()):
    these_results = main_results >> mask(X.question_conditions == question_condition)    
    # Convert complexity labels to more informative string labels.
    # Note that seaborn has problems with numbers as categories, 
    # and simple string conversion via `str` doesn't work.
    these_results["complexity"]  = these_results["complexity"].apply(lambda x: f"{x} x {x}")
    sns.lineplot(
        x="switch_conditions",
        y="mean_ratings",
        hue="complexity",
        data=these_results,
        ax=ax[i],
        linewidth=4
    )
    ax[i].set_title(f"{question_condition} Results")
    ax[i].set(xlabel='Switch Rate', ylabel='Mean Rating')    
plt.show()