numpy矩阵运算

import numpy as np

a_1 = np.array([10, 20, 30, 40])
b_1 = np.arange(4)
print(a_1, b_1)

# 数组相减
c_1 = a_1-b_1
print(c_1)

# 数组的乘方操作
d_1 = b_1**2
print(d_1)

# 计算正弦值
e_1 = 10*np.sin(a_1)
print(e_1)

# 数组的布尔操作（满足元素的返回True,不满足的返回False)
print(b_1)
print(b_1<3)

# 矩阵运算
f_1 = np.array([[1, 1],
                [0, 1]])
g_1 = np.arange(4).reshape((2, 2))
print(f_1, g_1)

# 对应元素相乘
h_1 = f_1*g_1
print(h_1)

# 矩阵乘法
h_2 = np.dot(f_1, g_1)
print(h_2)

h_3 = f_1.dot(g_1)
print(h_3)

print(g_1)
print(np.sum(g_1))  # 另种写法：print(g_1.sum())
print(np.min(g_1))
print(np.max(g_1))
print(np.mean(g_1))  # 指定axis为按列还是按行计算均值
print(np.average(g_1))
print(np.median(g_1))
print(np.cumsum(g_1))  # 累加，返回元素为之前元素的累加
print(np.nonzero(g_1)) # 返回两个矩阵描述非零元素的位置
g_2 = np.arange(14, 2, -1).reshape((3, 4))
print(np.sort(g_2))  # 逐行排序
print(np.transpose(g_1))  # 另外可写为：g_1.T
print(np.clip(g_1, 1, 2))  # 矩阵中小于1的为1，大于2的为2


print(np.sum(g_1, axis=1))  # 每行相加，返回列
print(np.sum(g_1, axis=0))  # 每列相加，返回行

i_1 = np.arange(12, 24).reshape((3, 4))
print(i_1)
print(np.argmax(i_1))  # 输出索引
print(np.argmin(i_1))

学习自莫凡Python