matplotlib - 基於附加數據的情節多個標記

Question

可以說我有以下數據說

x=[0,1,2,3,4,5,6,7,8,9] 
y=[1.0,1.5,2.3,2.2,1.1,1.4,2.0,2.8,1.9,2.0] 
z=['A','A','A','B','B','A','B','B','B','B'] 
plt.plot(x,y, marker='S') 

會給我有方形標記物的X-Y的情節。有沒有辦法根據數據z更改標記類型，以便所有'A'類型都有方形標記，'B'類型有三角形標記。

但我要添加在曲線上的「Z」的數據，只有當它（從「A」向「B」，或反之亦然在這種情況下）改變從一種類型到另

Answer 1

import matplotlib.pyplot as plt 

fig = plt.figure(0) 
x=[0,1,2,3,4,5,6,7,8,9] 
y=[1.0,1.5,2.3,2.2,1.1,1.4,2.0,2.8,1.9,2.0] 
z=['A','A','A','B','B','A','B','B','B','B'] 
plt.plot(x,y) 

if 'A' in z and 'B' in z: 
    xs = [a for a,b in zip(x, z) if b == 'A'] 
    ys = [a for a,b in zip(y, z) if b == 'A'] 
    plt.scatter(xs, ys, marker='s') 

    xt = [a for a,b in zip(x, z) if b == 'B'] 
    yt = [a for a,b in zip(y, z) if b == 'B'] 
    plt.scatter(xt, yt, marker='^') 
else: 
    plt.scatter(x, y, marker='.', s=0) 


plt.show() 

或者

import matplotlib.pyplot as plt 

fig = plt.figure(0) 
x=[0,1,2,3,4,5,6,7,8,9] 
y=[1.0,1.5,2.3,2.2,1.1,1.4,2.0,2.8,1.9,2.0] 
z=['A','A','A','B','B','A','B','B','B','B'] 
plt.plot(x,y) 

if 'A' in z and 'B' in z: 
    plt.scatter(x, y, marker='s', s=list(map(lambda a: 20 if a == 'A' else 0, z))) 
    plt.scatter(x, y, marker='^', s=list(map(lambda a: 20 if a == 'B' else 0, z))) 
else: 
    plt.scatter(x, y, marker='.', s=0) 


plt.show()