Как я могу создать скрипичный сюжет из списка данных, разделенных двоичным классом?

Question

Я хочу показать на скрипке сюжет о возрастах людей, каждый из которых принадлежит к 0 или 1 классу. Я создал список возрастов и отдельный список, соответствующий классу.Я могу построить сюжет скрипки для одного списка, но как я могу построить распределение по возрасту, разделенное на классы 1 и 0?

import numpy as np
import seaborn as sns
import csv
import matplotlib.pyplot as plt

    reader = csv.reader(file)

    ages = []
    class = []

    #Here we populate our list with data from our csv
    for column in reader:
        ages.append(column[3])
        class.append(column[0])

    #Here we can initialize Figure and Axes object
    fig, ax = plt.subplots()

    #Here we create our violin plot for the age distribution
    ax.violinplot(ages, vert=False)

    #
    # Here we need to add code to plot age distribution seperated by class
    #

    #Here we show our violin plot
    plt.show()

Answer 1

Я предполагаю, что вам нужно использовать параметр x или hue функции сюжета для скрипки морского волка

С вашими данными это будет выглядеть как

 #
# Here we need to add code to plot age distribution seperated by class
 sns.violinplot(x=class, y=ages)

Вы можете найтинекоторые примеры параметров hue и x в документации по seaborn: https://seaborn.pydata.org/generated/seaborn.violinplot.html