更改散點圖中不同虛擬值的點的顏色

Question

在我的數據集中，我有一個“ Price列用於房價，還有5個虛擬列用於城市中的不同位置。 我要做的是用不同的顏色在散點圖上顯示數據點。

例如，在一個包含所有房屋價格的散點圖上，我想要：

• 當dummy1指示房屋位於Area1時，所有價格點均為紅色。
• 藍色所有價格點時dummy2這表明房子是Area2等於2。

依此類推，直到最后一列。 如何創建該圖？ 我可以使用plt.scatter()創建沒有顏色的散點圖，但是不知道如何添加顏色代碼。

Answer 1

看一下matplotlib.pyplot.scatter的文檔，其中描述了參數c ，可以是

長度為n的顏色規范序列。

這是一個示例，該示例創建100個隨機的x和y數據點。 如果y值超過5 ，則該點將為藍色，否則為c列表中指定的紅色。

import matplotlib.pyplot as plt
import random

x = list(range(100))
y = [random.randint(0, 10) for _ in range(len(x))]
c = ["b" if y > 5 else "r" for y in y]

plt.scatter(x, y, c=c)
plt.show()

輸出將如下所示：