python散布図の色分けについて

Question

前提・実現したいこと

excelのデータをデータフレームにして、pythonにて散布図を作成しており、各プロット毎に色分けを行いたいのですが上手くいきませんのでご教示願います。

no=x軸
place=y軸
fruit=色分け（色は自動で凡例が欲しいです。）

テスト用データ

fruit no place
apple 1 100
apple 2 120
apple 3 140
orange 4 80
orange 5 60
peach 6 150
peach 7 180
peach 8 200

該当のソースコード

import pandas as pd
import matplotlib.pyplot as plt
from pandas.plotting import scatter_matrix

file=r"C:\Users\test\Desktop\個人用\fruit.xlsx"

df=pd.read_excel(file,sheet_name="fruit")

plt.scatter(df['no'], df['値段'])
plt.ylabel('値段')
plt.xlabel('no')
plt.grid()
plt.show()

試したこと

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[tachikoma]

テストデータのplaceはpriceのタイポですかね。y軸に値段と設定してるので。

[kuusan]

タイポです。すみません。ｙ軸がpriceです。

Answer 1

こんな感じでよいのではないでしょうか

df['fruit'] = df['fruit'].astype('category')
plt.scatter(df['no'], df['price'], c=df['fruit'].cat.codes)

---

【補足】
色指定＆凡例を出す場合

colors = {'orange':'orange','apple':'red','peach':'pink'}
for f in df['fruit'].unique():
    plt.scatter(df.loc[df.fruit == f ,'no'], df.loc[df.fruit == f ,'値段'], c=colors[f], label=f)
plt.legend()
plt.show()

---

【更に】

特に色を指定しないなら

for f in df['fruit'].unique():
    plt.scatter(df.loc[df.fruit == f ,'no'], df.loc[df.fruit == f ,'値段'], label=f)
plt.legend()
plt.show()

Comments

[kuusan]

すみません。大体そんな感じで、後色と凡例を合わせたいのですが可能でしょうか？

[magichan]

凡例を出すならばループ使ったほうが楽かも。

とりあえず補足を書きましたのでご確認ください

[kuusan]

ありがとうございます！
最後に色は指定ではなく、ランダム（隣の色と被らないような色）で表示させたいのですがご教示頂いてもよろしいでしょうか？

[magichan]

一応　色を指定しない方法も追加

[kuusan]

本当にありがとうございました！
助かりました！

Answer 2

気の利いた色の組み合わせを得たいときはこちらが参考になりそうです。
https://qiita.com/SaitoTsutomu/items/c79c9973a92e1e2c77a7

Comments

[kuusan]

ありがとうございます。参考にしてみます。

Answer 3

plt.scatter()の引数に色を渡せばできます。
以下を参照してください。
https://qiita.com/supersaiakujin/items/ef85dd247cefdfbd2e2a

http://sinhrks.hatenablog.com/entry/2015/11/15/222543