【Pandas】データフレームからクラスごとに色分けされた散布図を2行で作成する方法

前提

以下のようなPandasのデータフレームがあったとします。

# pandasをimportしていること前提
# dfはpandas.DataFrame（100 rows × 3 columns）
df.head()

	y	x	f
0	6	8.31	C
1	6	9.44	C
2	6	9.50	T
3	12	9.07	C
4	10	10.16	T

ここから列fのクラス(C/T)ごとに色分けした凡例付きの散布図を作成したい。

方法

以下の2行のコードで簡単に実現できます（Python 3.6で確認）。

import seaborn as sns
sns.scatterplot(x='x', y='y', hue='f', data=df)

f:id:mimosom:20190501111219p:plain

ラクですね。

同じことは、以下のようなコードでも実現できます*1。

import matplotlib.pyplot as plt
fig, ax = plt.subplots()
for name, group in df.groupby('f'):
ax.plot(group.x, group.y, marker='o', linestyle='', ms=8, label=name)
ax.legend()
plt.show()

f:id:mimosom:20190501102629p:plain

他にもplt.scatter()でx, yにクラスごとのデータを代入する方法など色々あるかと思いますが、拘りがなければseabornのscatterplot()が一番楽かと思います。

以上です。

*1:参考：python - Scatter plots in Pandas/Pyplot: How to plot by category - Stack Overflow

プログラミング言語はPythonがおすすめな4つの理由【効率的な独学方法も紹介します】

どのプログラミング言語を学べばいいか分からない。Pythonに入門しようか迷っている。効率良くPythonを独学したい。この記事では現役エンジニアで日頃からPythonを使う筆者が、Pythonの魅力と効率的な独学方法をご紹介します。