Python matplotlib - Как нарисовать линейный график со многими сериями?

Question

У меня был кадр данных Python Pandas, как показано ниже:

    test_score  param   # of Nodes
0   0.497852    relu        1
1   0.68935     relu        2
2   0.701165    relu        3
3   0.735223    identity    1
4   0.735064    identity    2
5   0.735691    identity    3
6   0.72651     logistic    1
7   0.664837    logistic    2
8   0.743445    logistic    3
9   0.705182    tanh        1
10  0.673399    tanh        2
11  0.684129    tanh        3

Я хотел бы нарисовать линейную диаграмму с осью x как "# of Nodes", осью y как "test_score" и четырьмя значениями параметров "relu, identity, logistic, tanh" как линии 4 серий.

Может ли эта диаграмма быть построена в python matplotlib?

Answer 1

Если у вас есть все узлы для всех параметров, как в вашем примере, вы можете .pivot ваш DataFrame в более подходящий формат для построения их всех.

df.pivot(index='# of Nodes', columns='param', values='test_score').plot()

Answer 2

Вы можете начать с группировки по param, затем итерацией по вашим группам и составлением графика:

g = df.groupby('param')

for p, data in g:
    plt.plot(data['# of Nodes'], data['test_score'], label=p)

plt.legend()
plt.xlabel('# of Nodes')
plt.ylabel('Test Score')

plt.show()