CODEWITHSUNDEEP
pythonmatplotlib
rambalachandran
rambalachandran

Reputation: 2201

matplotlib - multiple markers for a plot based on additional data

Lets say I have the following data say

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')

will give me a x-y plot with square markers. Is there a way to change the marker type based on the data z, so that all 'A' type has square marker and 'B' type has a triangle marker.

but I want to add 'z' data on the curve only when it changes from one type to another (in this case from 'A' to 'B' or vice versa)

Upvotes: 2

Views: 2807

Answers (1)

Leon
Leon

Reputation: 32474

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()

Or

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()

Upvotes: 1

Related Questions