Python numpy的基础运算2

一、矩阵最大值最小值的索引

import numpy as np

a=np.arange(2,14).reshape((3,4))

print(np.argmin(a))       #最小值的索引
print(np.argmax(a))       #最大值的索引
print(a)

输出：

0
11
[[ 2  3  4  5]
 [ 6  7  8  9]
 [10 11 12 13]]

二、矩阵求平均值

import numpy as np

a=np.arange(2,14).reshape((3,4))

print(np.mean(a))
print(a.mean())
print(np.average(a))

输出：

7.5
7.5
7.5

三、前n项和

import numpy as np

a=np.arange(2,14).reshape((3,4))

print(np.cumsum(a))

输出：

[ 2  5  9 14 20 27 35 44 54 65 77 90]

四、累差

import numpy as np

a=np.arange(2,14).reshape((3,4))

print(np.diff(a))

输出：

[[1 1 1]
 [1 1 1]
 [1 1 1]]

五、输出索引

import numpy as np

a=np.arange(2,14).reshape((3,4))

print(np.nonzero(a))

输出：

(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))

六、排序和专制

import numpy as np

a=np.array(
    [[3,4,5],
    [8,9,7],
    [1,3,2]]
)

print(a)
print(np.sort(a))
print(np.sort(a,axis=1))  #默认按行排序
print(np.sort(a,axis=0))  #按列排序
print(np.transpose(a))    #转置

输出：

[[3 4 5]
 [8 9 7]
 [1 3 2]]
[[3 4 5]
 [7 8 9]
 [1 2 3]]
[[3 4 5]
 [7 8 9]
 [1 2 3]]
[[1 3 2]
 [3 4 5]
 [8 9 7]]
[[3 8 1]
 [4 9 3]
 [5 7 2]]