how to generate 2 clusters of scatters randomly with numpy or some other packages in Python?

Question

I am trying to draw a plot, like following (call it img_1) , programmatically.

this post gives a piece of code though, the key is iris dataset

is there a way to generate 2 group of data randomly with NumPy or some other packages to draw a scatter plot like img_1?

Answer 1

Use np.random to generate the point clouds and matplotlib to plot them.

import numpy as np
import matplotlib.pyplot as plt

# generate random normally distributed point clouds (customize as needed)
x1 = np.random.normal(loc=3.0, size=15)
y1 = np.random.normal(loc=2.0, size=15)
x2 = np.random.normal(loc=9.0, size=35)
y2 = np.random.normal(loc=7.0, size=35)

# now do the plotting with matplotlib
plt.scatter(x1,y1,color='red', marker='+',s=35)
plt.scatter(x2,y2,color='blue', marker= '^',s=35)
plt.xlim(-5,15)
plt.ylim(-5,15)
plt.xlabel('$x_1$',fontsize=25)
plt.ylabel('$x_2$',fontsize=25)
plt.savefig('example.png', bbox_inches='tight')
plt.show()

This generates