Python 之 Numpy (三)运算

代码：

import numpy as np

# 加减
a = np.array([10, 20, 30, 40])
b = np.arange(4)
c = a - b
print(a, b)
print(c)

运行结果：

[10 20 30 40] [0 1 2 3]
[10 19 28 37]

代码：

# 三角函数
a = np.array([10, 20, 30, 40])
b = 10 * np.sin(a)
print(b)

运行结果：

[-5.44021111  9.12945251 -9.88031624  7.4511316 ]

代码：

# 判断矩阵里元素
a = np.array([10, 20, 30, 40])
b = np.arange(4)
print(b)
print(b < 3)
print(b == 3)

运行结果：

[0 1 2 3]
[ True  True  True False]
[False False False  True]

代码：

# 二维矩阵运算
a = np.array([[1, 1],
              [0, 1]])
b = np.arange(4).reshape((2, 2))
print(a)
print(b)
# 乘法
# 逐个相乘
c = a * b
print(c)
# 矩阵乘法
c = np.dot(a, b)
c1 = a.dot(b)
print(c)
print(c1)

运行结果：

[[1 1]
 [0 1]]
[[0 1]
 [2 3]]
[[0 1]
 [0 3]]
[[2 4]
 [2 3]]
[[2 4]
 [2 3]]

代码：

# 随机生成矩阵
a = np.random.random((2, 4))
print("随机生成矩阵")
print(a)
print(np.sum(a, axis=1))  # 按行求和
print(np.min(a, axis=0))  # 按列
print(np.max(a, axis=1))  # 按行

运行结果：

随机生成矩阵
[[ 0.50727226  0.69861239  0.4337988   0.94719061]
 [ 0.52180977  0.33662019  0.36415968  0.37560842]]
[ 2.58687406  1.59819807]
[ 0.50727226  0.33662019  0.36415968  0.37560842]
[ 0.94719061  0.52180977]

代码：

# 索引
A = np.arange(2, 14).reshape((3, 4))
print(A)
# 最小值索引
print(np.argmin(A))
# 最大值索引
print(np.argmax(A))
# 平均值
print(A.mean())
# 中位数
print(np.median(A))
# 累加
print(np.cumsum(A))
# 累差
print(np.diff(A))
# 找出非零数
print("非零数")
print(np.nonzero(A))
# 排序
print(np.sort(A))
# 矩阵的转置
print(A.T)
print(np.transpose(A))
print((A.T).dot(A))
# 所有小于5的数都等于5，所有大于9的数都等于9
print(A)
print(np.clip(A, 5, 9))

运行结果：

[[ 2  3  4  5]
 [ 6  7  8  9]
 [10 11 12 13]]
0
11
7.5
7.5
[ 2  5  9 14 20 27 35 44 54 65 77 90]
[[1 1 1]
 [1 1 1]
 [1 1 1]]
非零数
(array([0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2]), array([0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3]))
[[ 2  3  4  5]
 [ 6  7  8  9]
 [10 11 12 13]]
[[ 2  6 10]
 [ 3  7 11]
 [ 4  8 12]
 [ 5  9 13]]
[[ 2  6 10]
 [ 3  7 11]
 [ 4  8 12]
 [ 5  9 13]]
[[140 158 176 194]
 [158 179 200 221]
 [176 200 224 248]
 [194 221 248 275]]
[[ 2  3  4  5]
 [ 6  7  8  9]
 [10 11 12 13]]
[[5 5 5 5]
 [6 7 8 9]
 [9 9 9 9]]