genNew.py (444B)
1 import matplotlib.pyplot as plt 2 import seaborn as sns 3 import random 4 5 x = [] 6 y = [] 7 target = [] 8 9 for i in range(0,1000): 10 rnd = (random.random() * 300) + 100 11 rndAdd = rnd + (random.random() * 140 - 70) 12 x.append(rndAdd) 13 height = (random.random() * 40 + 30) + rndAdd * 0.1 14 y.append(height) 15 16 if rnd / height > 4: 17 target.append(1) 18 else: 19 target.append(0) 20 plt.scatter(y,x, c=target, cmap='bwr') 21 22 plt.show()