增加lab2.py

.gitignore

@@ -102,3 +102,6 @@ venv.bak/
 
 # mypy
 .mypy_cache/
+
+# pycharm .idea
+.idea/

lab2/lab2.py

@@ -0,0 +1,52 @@
+import numpy as np
+from matplotlib import pyplot as plt
+from scipy.stats import binom
+
+""" TODO
+    1.首先不生成数据，利用已有的数据(如西瓜书上的例子)先进行测试
+    2.尽量使用lab1中已经造好的轮子进行
+ """
+
+
+# def generateData(number, meanPos, meanNeg, scale1=0.1, scale2=0.2):
+#     y = np.random.randint(0, 2, (number, 1))
+#     meanPos_1 = meanPos * np.random.random_sample()
+#     meanPos_2 = meanPos * np.random.random_sample()
+#     meanNeg_1 = meanNeg * np.random.random_sample()
+#     meanNeg_2 = meanNeg * np.random.random_sample()
+#     x_1 = []
+#     x_2 = []
+#     for i in range(number):
+#         x_1.append(np.random.)
+#     x_1 = np.random.normal(loc=mean_1, scale=scale1, size=(number, 1))
+#     x_2 = np.random.normal(loc=mean_2, scale=scale2, size=(number, 1))
+#     return x_1, x_2, y
+
+
+# number = 20
+# mean_1 = 2
+# mean_2 = -2
+# x_1, x_2, y = generateData(number, mean_1, mean_2)
+# print(x_1.T)
+# print(x_2.T)
+
+# type1_x = []
+# type1_y = []
+# type2_x = []
+# type2_y = []
+# for i in range(number):
+#     if y[i] == 0:
+#         type1_x.append(x_1[i])
+#         type1_y.append(x_2[i])
+#     else:
+#         type2_x.append(x_1[i])
+#         type2_y.append(x_2[i])
+
+# plt.scatter(type1_x, type1_y, facecolor="none", edgecolor="b", label="positive")
+# plt.scatter(type2_x, type2_y, marker="x", c="r", label="negative")
+# plt.legend()
+# plt.show()
+
+# # k = np.arange(0, number + 1)
+# # binomial = binom.pmf(k, number, p)
+# # print(binomial)