包引入
import numpy as np
import matplotlib.pylab as plt
from mpl_toolkits.mplot3d import Axes3D
问题1.1
A = np.eye(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 样式)顺序存储在内存中。
问题2
根据城市人口数量,预测开小吃店的利润 数据在ex1data1.txt里,第一列是城市人口数量,第二列是该城市小吃店利润。
2.1
读入数据,然后展示数据
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)
|