python - 使用 matplotlib 将另一列作为类别绘制特定列值

Question

我想绘制数据集中特定列的线图，每个类别一条线，该类别是另一列。

1)让我们从生成一个假数据集开始

import pandas

import random

df = pandas.DataFrame(np.random.randn(100, 7), columns=list('ABCDEFG'))
df['Z'] = np.random.choice(range(1, 6), df.shape[0])

df

(想发布输出图片，但我没有任何声誉，所以让我知道如何弥补)

2) 我的问题是我想要一个 x 轴为 1 到 100 的图，y 轴显示变量 B、C 和 D 值，每个 Z 变量类别(1 到 6 之间)有一行并带有图例为此。

非常感谢

编辑:我找到了一个解决方案，但我认为它不是很优雅

我首先将行转置为列

df.set_index('Z').T

然后我使用

进行绘图

df[['B','C','D']].plot(figsize=(10,5))

但是我不认为这是一个优雅的解决方案？ (我对Python真的很陌生)。因此，如果有人对原始数据帧的绘图有一个很好的答案，我会很高兴看到它!

Answer 1

正如我的编辑中所述，解决方案如下:

1)首先将我的行转置为列

df.set_index('Z').T

2)然后使用绘图

df[['B','C','D']].plot(figsize=(10,5))