CODEWITHSUNDEEP
pythonpython-3.xpandasmatplotlib
Boom
Boom

Reputation: 1315

How to create scatter plot with 3 columns

I have a df which looks:

df_results = pd.DataFrame(data={})
df_results['names'] = ['James', 'Lucas', 'Henry', 'James', 'Lucas', 'Henry']
df_results['try_name'] = ["try_1", "try_1", "try_1", "try_2", "try_2", "try_2"]
df_results['score'] = [0.7, 0.9, 0.3, 0.91, 0.1, 0.2]

   names try_name  score
0  James    try_1   0.70
1  Lucas    try_1   0.90
2  Henry    try_1   0.30
3  James    try_2   0.91
4  Lucas    try_2   0.10

I want to create a plot (scatter) where:

How can I create this plot ?

Upvotes: 2

Views: 2203

Answers (2)

Guillaume Ansanay-Alex
Guillaume Ansanay-Alex

Reputation: 1252

Pure matplotlib solution:

fig = plt.figure(figsize=(7,4))
ax = fig.add_subplot(111)
for name in df_results.names.unique():
    ax.scatter(df_results[df_results.names == name].try_name, 
               df_results[df_results.names == name].score, label=name)
ax.legend()

Using pandas plot:

fig = plt.figure(figsize=(7,4))
ax = fig.add_subplot(111)
for i,name in enumerate(df_results.names.unique()):
    df_results[df_results.names == name].plot.scatter('try_name', 'score', 
                                                      ax=ax, color='C{}'.format(i), 
                                                      label=name)
ax.legend()

Upvotes: 2

tdy
tdy

Reputation: 41327

You could use seaborn's swarmplot(), stripplot(), or scatterplot().

For example with swarmplot():

import seaborn as sns
sns.swarmplot(data=df_results, x='try_name', y='score', hue='names')

swarmplot example

Upvotes: 2

Related Questions