(data, title)
| 44 | |
| 45 | #---------可视化----------------- |
| 46 | def plot_data(data, title): |
| 47 | color = [] |
| 48 | for i in data[:, 2]: |
| 49 | if i == 0: |
| 50 | color.append("orange") |
| 51 | else: |
| 52 | color.append("blue") |
| 53 | |
| 54 | plt.scatter(data[:, 0], data[:, 1], c=color) |
| 55 | plt.title(title) |
| 56 | plt.show() |
| 57 | |
| 58 | # ----------------------- |
| 59 | if __name__ == "__main__": |