import numpy as np
# 计算1到100的平方和
a = np.arange(1, 101)
sum_of_squares = np.sum(a**2)
print(sum_of_squares) # 输出:338350
# 计算sin函数在0到2π范围内的取值
x = np.arange(0, 2*np.pi, 0.1)
y = np.sin(x)
print(y) # 输出:[0. 0.09983342 0.19866933 0.29552021 0.38941834 ...]
5.2. 数组的创建和初始化
arange函数可以用于创建和初始化数组。
import numpy as np
# 创建一个长度为10的零向量
a = np.zeros(10)
# 创建一个长度为10的单位矩阵
b = np.eye(10)
# 创建一个长度为5的全1向量
c = np.ones(5)
# 创建一个长度为5的随机向量
d = np.random.rand(5)