How to add lines from 0,0 to each scatter plot point using matplotlib?

Question

I'm trying to recreate the following chart with matplotlib, but I can't figure it out how to create the lines from the 0,0(origin) to each point

my current code is:

plt.figure(figsize=(10,7))
plt.grid()
plt.xlabel('Movie 1')
plt.ylabel('Movie 2')
A = [1.0, 2.0]
B = [2.0, 4.0]
C = [2.5, 4.0]
D = [4.5, 5.0]
xs = [A[0], B[0], C[0], D[0]]
ys = [A[1], B[1], C[1], D[1]]

users = ['A', 'B', 'C', 'D']

for i, user in enumerate(users):
    x = xs[i]
    y = ys[i]
    plt.scatter(x, y, marker = 'o')
    plt.plot(x,y,0,0, linestyle = '--' )
    plt.text(x+0.01, y+0.01, user, fontsize=19)
    
plt.xlim(0,7)
plt.ylim(0,7)


plt.show()

The code only returns the points, but with no lines... how do I create the lines?

Thanks!

To plot the A line using plt.plot(x,y...: x is (0,1) and y is (0,2). You can zip (0,0) with A to make those. Same with the other three. Pyplot Tutorial. — wwii
– wwii, Commented Dec 10, 2020 at 19:34

Lucas Roy · Accepted Answer · 2020-12-10 19:43:42Z

1

you can do this by plotting the lines manually in your for loop:

for i, user in enumerate(users):
    x = xs[i]
    y = ys[i]
    plt.scatter(x, y, marker = 'o')

    # V this adds the lines V
    plt.plot([0,xs[i]], [0, ys[i]], color="black")
    # ^ this adds the lines ^

    plt.plot(x,y,0,0, linestyle = '--' )
    plt.text(x+0.01, y+0.01, user, fontsize=19)

which results in this graph

answered Dec 10, 2020 at 19:43

Lucas Roy

1331 silver badge7 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to add lines from 0,0 to each scatter plot point using matplotlib?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related