吴恩达机器学习笔记——练习1

import numpy as np
import matplotlib.pylab as plt
from mpl_toolkits.mplot3d import Axes3D

A = np.eye(5)#打印维度为5的方阵
print(A)

np.eye (N, M=None, k=0, dtype=float, order=‘C’) -> Any
作用：返回一个矩阵，对角线上为1，其他地方为0
参数N：几行
参数M：几列，如果没有定义，就和N一样
参数K：矩阵的对角线平移，正数则向右平移，负数则向左平移。
参数dtype：返回的数组的数据类型
order ：没什么用，输出是否应以行优先（C 样式）或列优先（Fortran 样式）顺序存储在内存中。

根据城市人口数量，预测开小吃店的利润
数据在ex1data1.txt里，第一列是城市人口数量，第二列是该城市小吃店利润。

读入数据，然后展示数据

# 绘制散点图
def plotData(x, y):
    plt.plot(x, y, 'rx', ms=10)
    plt.xlabel('Population of City in 10,000')
    plt.ylabel('Profit in $10,000')
    plt.show()

print('Plotting Data...')
data = np.loadtxt('ex1data1.txt', delimiter=',')
X = data[:, 0]; Y = data[:, 1]
m = np.size(Y, 0)
plotData(X, Y)