numpy 矩阵操作

import numpy as np

# 定义一个矩阵
A = np.arange(14,2, -1).reshape((3,4))
print(A)

# 求矩阵A最小值索引
print(np.argmin(A))
# 求矩阵A最大值索引
print(np.argmax(A))
# 求矩阵平均值
print(np.mean(A))
print(A.mean())
print(np.average(A))
# 求中位数
print(np.median(A))
# 累加
print(np.cumsum(A))
# 累差
print(np.diff(A))
# 找出非零的数
print(np.nonzero(A))
# 排序
print(np.sort(A))
# 矩阵的转置
print(np.transpose(A))
print((A.T).dot(A))
# 在数组A中所有小于5的数都修改成5，所有大于9的数都修改成9
print(np.clip(A, 5, 9))
# 对列求平均值
print(np.mean(A,axis=0))

结果：

[[14 13 12 11]
[10 9 8 7]
[ 6 5 4 3]]
11
0
8.5
8.5
8.5
8.5
[ 14 27 39 50 60 69 77 84 90 95 99 102]
[[-1 -1 -1]
[-1 -1 -1]
[-1 -1 -1]]
(array([0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2], dtype=int64), array([0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3], dtype=int64))
[[11 12 13 14]
[ 7 8 9 10]
[ 3 4 5 6]]
[[14 10 6]
[13 9 5]
[12 8 4]
[11 7 3]]
[[332 302 272 242]
[302 275 248 221]
[272 248 224 200]
[242 221 200 179]]
[[9 9 9 9]
[9 9 8 7]
[6 5 5 5]]
[10. 9. 8. 7.]