对于列中的每个唯一元素,创建一个具有其对应特征的图形

时间:2019-09-03 12:10:58

标签: python pandas dataframe

我相信我的问题确实很简单,并且必须有一种非常简单的方法来解决我仍然不知道的熊猫问题。我真的不想创建新的数据框。我已经花了一些时间试图自己弄清楚,但无法取得结果。

一个简单的示例可能是一个更好的方法来解释我的目标:在C列上,我有四个不同的唯一值,如下面的代码所示。 (值_1,值_2,值_3,值_4),绘制一个图形,其中“ x”是该值在列A中的对应特征,而“ y”是对应的列B。

import pandas as pd
data = {'Column A': [100,200,300,400,500,500,500,300],
    'Column B': [1,1,2,2,3,3,0,2], 
    'Column C': ["Value_1", "Value_2", "Value_3", "Value_4", "Value_1", "Value_2", "Value_3", "Value_4"]}
df = pd.DataFrame(data, columns=['Column A','Column B', 'Column C'])

之后我希望得到一个结果,因为在这种情况下,我将收到四个图,每个图代表C列的值之一以及A和B列的对应特征。

我尝试了以下方法:

for d in df.groupby(df['Column C']):
df.plot.scatter(x='Column A', y='Column B', marker='.')
plt.show()

但是,这样做时,由于没有在C列中实现的过滤功能,因此我无法真正实现所需的功能,因为所有点都绘制在该图中。

希望我能简洁明了。祝您一切顺利,并感谢您的关注。

1 个答案:

答案 0 :(得分:1)

我想您可以使用以下方法遍历唯一值:

for val in df['Column C'].unique():
    sdf = df.loc[df['Column C']==val]
    sdf.plot(x='Column A', y='Column B', label='C = {}'.format(val))