8.基础绘图

最新推荐文章于 2021-06-08 18:56:11 发布

原创最新推荐文章于 2021-06-08 18:56:11 发布 · 305 阅读

0 ·

CC 4.0 BY-SA版权

机器学习专栏收录该内容

17 篇文章

订阅专栏

本文详细介绍了一种使用Python的数据可视化方法，通过matplotlib库绘制离散点图的过程。具体包括如何从CSV文件读取数据，区分正例和负例，并用不同颜色和标记展示在图表上，最终呈现考试成绩与录取情况的关系。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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()