8.基礎繪圖

1.離散點圖

代碼：

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt


path = 'LogiReg_data.txt'
data = pd.read_csv(path, header=None, names=['Exam1', 'Exam2', 'Admitted'])
# 正例
positive_data = data[data['Admitted'] == 1]
# 負例
negative_data = data[data['Admitted'] == 0]
# 繪圖
fig, ax = plt.subplots(figsize=(10, 5))
ax.scatter(positive_data['Exam1'], positive_data['Exam2'], s=30, c='b', marker='o', label='Admitted')
ax.scatter(negative_data['Exam1'], negative_data['Exam2'], s=30, c='r', marker='x', label='Not Admitted')
ax.legend()
ax.set_xlabel('Exam 1 Score')
ax.set_ylabel('Exam 2 Score')
fig.show()

效果