numpy基础运算

import numpy as np

a = np.array([10, 20, 30, 40])
b = np.arange(4)
# 使用a+b，逐个元素对应相加
c = a + b
print("# 使用a+b，逐个元素对应相加:\n a:[10, 20, 30, 40]\nb:[1,2,3,4]\n", c, end="\n\n")

# 三角函数使用：sin,cos
d = np.sin(a)
print("对a:" + str(a) + "的每个元素求sin:\n", d)

# 可以直接使用数字与列表比较大小，会返回列表，列表的每个元素表示是否满足条件

res = 3 > b
print(b)
print("b 是[0,1,2,3],用3>b进行比较，返回：\n", res, end="\n\n")

e = np.array([[1, 2],
              [3, 4]])
f = np.array([[1, 2],
              [1, 2]])
# 使用* 返回一一对应乘法结果，使用np.dot(e,d)返回矩阵乘法
g = e * f
h = np.dot(e, f)
# 也可以用a.dot(b)
i = e.dot(f)
print("# 使用* 返回一一对应乘法结:\n", g, end="\n\n")
print("# 使用np.dot(e,d)返回矩阵乘法\n", h, end="\n\n")
print("# 使用e.dot(f)返回矩阵乘法\n", i, end="\n\n")

# 随机创建一个array,元素大小为介于0到1之间，需要指定shape
j = np.random.random((2, 3))
print("使用np.random.random((2, 3))随机创建一个array,元素大小为介于0到1之间，需要指定shape:\n", j, end="\n\n")

# 求最大值，最小值，求和
all_max = np.max(e, axis=1)  # axis=0 按列求最大值 axis=1 按行求最大值
all_min = np.min(e, axis=1)  # axis=0 按列求最小值 axis=1 按行求最小值
all_sum = np.sum(e, axis=1)  # axis=0 按列求和 axis=1 按行求和
print(all_max)
print(all_min)
print(all_sum)