Python 基於 list2 中的項目對 list1 中的項目進行分組

Question

我有兩個列表，其中一個有類似的點

list1 = [1.8, 4.5, 1.1, 2.1, 9.8, 7.6, 11.32, 3.2, 0.5, 6.5] 

另一個根據用戶輸入的內容從 0 到 x 不等。 類似的東西

list2 = [0, 1, 2, 0, 4, 4, 4, 3, 2, 1]

我正在嘗試根據它們的顏色使用散點圖 plot 繪制它們

plt.scatter(list1, list2)
plt.show()

那就是按照我的意願繪制它們，但它們都是相同的顏色。 我知道要切換顏色我會做

plt.scatter(list1, list2, color="red")

但這會將每個點都變成紅色。 我想說的是，對於 list2 中的每個點，更改 list1 中相同索引處的點顏色，然后繪制它。

---

import matplotlib.pyplot as plt    
list1 = [1.8, 4.5, 1.1, 2.1, 9.8, 7.6, 11.32, 3.2, 0.5, 6.5] 
list2 = [0, 1, 2, 0, 4, 4, 4, 3, 2, 1]
plt.scatter(list1, list2, color="red")
plt.show()

Answer 1

您可以使用數組作為顏色：

import matplotlib.pyplot as plt

list1 = [1.8, 4.5, 1.1, 2.1, 9.8, 7.6, 11.32, 3.2, 0.5, 6.5]
list2 = [0, 1, 2, 0, 4, 4, 4, 3, 2, 1]
plt.scatter(list1, list2, c=list2)

output：