循环散点图颜色

时间:2017-05-13 05:04:32

标签: python loops matplotlib colors scatter-plot

如何将散点图颜色循环? 我的代码:

col = {'Male':'green','Female':'blue'}

gender = [‘Male’,’Female’,’Male’,’Male’,’Female’, …]

Matched_Days = [list of days…]

Marital_Status = [list of statuses…]

for type in gender:

plt.scatter(Marital_Status, Matched_Days, c=col[type])

我只得到一种颜色:蓝色,因为最后一个性别是列表中的“女性”。

出于某种原因,我无法循环并注册字典中的所有颜色

1 个答案:

答案 0 :(得分:1)

您未正确使用matplotlib。您只需要一个分散,而不是一个while循环。

gender = [‘Male’,’Female’,’Male’,’Male’,’Female’, …]
gender_color=[]
for elem in gender:
  if elem=="Male":
    gender_color.append("green")
  else:
    gender_color.append("blue")
Matched_Days = [list of days…] 
Marital_Status = [list of statuses…]   
plt.scatter(Marital_Status, Matched_Days, c=gender_color)
plt.show()

c参数可以采用颜色列表。除非你想要多个图,否则你不应该使用for循环。