How to efficiently scatter plot a numpy 2d array

Question

I have a numpy with each row containing x, y pairs and I want to display a scatter plot without using a for loop so I used the following approach using pandas:

def visualize_k_means_output(self, centroids):
    fig, ax = plt.subplots()
    self.visualize_box_relative_sizes()
    frame = pd.DataFrame(centroids, columns=['X', 'Y'])
    ax.scatter(frame['X'], frame['Y'], marker='*', s=200, c='black')

The question is how to extract the first item as x and the second item as y without using a for loop for example:

ax.scatter(x=[item[0] for item in centroids], y=[item[1] for item in centroids], ...)

Bruno Mello · Accepted Answer · 2020-03-27 17:20:45Z

1

If I understood correctly, you want to slice your numpy array:

x = centroids[:, 0]
y = centroids[:, 1]

answered Mar 27, 2020 at 17:20

Bruno Mello

4,6781 gold badge16 silver badges46 bronze badges

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

Collectives™ on Stack Overflow

How to efficiently scatter plot a numpy 2d array

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