如何使一個散點圖，通過顏色matplotlib.pyplot分離

Question

我想要的情節一樣，...

有3個condistions ..

1. 如果x > 700 - > COL = '藍色'
2. 如果x < = 700且y> -2 - > COL = '黑'
3. 如果x < = 700和y < = -2 - > col ='red'and marker ='*'

我該如何做一個這樣的情節？

Answer 1

我沒有你的確切數據，所以我只是要創建自己的數據集。最簡單的方法是將三種不同的顏色類型存儲在三組獨立的數組中。檢查下面的代碼。

import matplotlib.pyplot as plt 
import random 

x_val = [random.randint(600, 800) for i in range(500)] 
y_val = [random.randint(-4, 0) for i in range(500)] 
blue_plot = [(x, y) for (x, y) in zip(x_val, y_val) if x > 700] 
black_plot = [(x, y) for (x, y) in zip(x_val, y_val) if x <= 700 and y >= -2] 
red_plot = [(x, y) for (x, y) in zip(x_val, y_val) if x <= 700 and y <= -2] 

def separate_x_y(val): 
    x = [x for (x, _) in val] 
    y = [y for (_, y) in val] 
    return x, y 

blue_x, blue_y = separate_x_y(blue_plot) 
black_x, black_y = separate_x_y(black_plot) 
red_x, red_y = separate_x_y(red_plot) 

# Main part of code 
plt.scatter(blue_x, blue_y, color = 'blue') 
plt.scatter(black_x, black_y, color = 'black') 
plt.scatter(red_x, red_y, color = 'red', marker = '*') 
plt.xlim(550, 850) 
plt.ylim(-6, 2) 
plt.show() 

你會得到這樣的情節。