Matplotlibの散布図の描画が遅いのですが、よい方法はありますか？

Question

問題

以下のようなDataFrameがあったとします。

index	都道府県	面積	価格
0	東京	100	100
1	東京	130	120
2	東京	90	100
3	埼玉	100	80
4	埼玉	120	80
5	埼玉	100	90
6	千葉	80	90
7	千葉	90	130
8	千葉	100	100

このDataFrameでx=面積、y=価格で散布図を作成したいです。
ただし、東京の場合は、赤くマーカーをつけたいです。

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

pref = ["東京", "東京", "東京", "埼玉", "埼玉", "埼玉", "千葉", "千葉", "千葉"]
s2 = [100, 130, 90, 100, 120, 100, 80, 90, 100]
value = [100, 120, 100, 80, 80, 90, 90, 130, 100]
df_t = pd.DataFrame({'都道府県':pref, '面積': s2, '価格':value})

index_list = df_t[df_t["都道府県"]=="東京"].index

x = np.array(df_t["面積"])
y = np.array(df_t["価格"])

for i in range(len(x)):
    plt.scatter(x[i], y[i], color='red') if i in index_list else plt.scatter(x[i], y[i], color='black')
plt.show()

そこでこのようなコードを作成しました。

データ数が少ない場合良いのですが、データ数を10万以上で行うと描画までかなりの時間がかかります。10分以上待ってもダメでした。

このような大規模な場合に一つのグラフに描画したい場合どうすればよいのでしょうか。
わかる方教えていただきたいです。

Answer 1

以下のように東京とそれ以外で描画すればよいかと思います。

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

pref = ["東京", "東京", "東京", "埼玉", "埼玉", "埼玉", "千葉", "千葉", "千葉"]
s2 = [100, 130, 90, 100, 120, 100, 80, 90, 100]
value = [100, 120, 100, 80, 80, 90, 90, 130, 100]
df_t = pd.DataFrame({'都道府県':pref, '面積': s2, '価格':value})

tko = df_t["都道府県"]=="東京"

df_tko = df_t[tko]
df_ntk = df_t[tko != True]

plt.scatter(df_tko['面積'], df_tko['価格'], color='red')
plt.scatter(df_ntk['面積'], df_ntk['価格'], color='black')
plt.show()

Comments

[WatanabeJin]

for分にするかしないかで一瞬で処理が終わりました。ありがとうございました。