numpy入门——中级计算

中级计算

-导入numpy 并以np的形式简写

import numpy as np
# 创建一个随机数组
a = np.arange(0,12).reshape((3,4))
print(a)

array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

• 计算a的中位数

print(np.median(a))

5.5

• 计算累加值

print(np.cumsum(a))

array([ 0,  1,  3,  6, 10, 15, 21, 28, 36, 45, 55, 66], dtype=int32)

• 计算累差值

print(np.diff(a))

array([[1, 1, 1],
       [1, 1, 1],
       [1, 1, 1]])

• 计算非零数的行数与列数

print(np.nonzero(a))

(array([0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2], dtype=int64),
 array([1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3], dtype=int64))

• 排序

a = np.arange(12,0,step = -1).reshape((3,4))
print(a)
# 默认按照每一行进行排序
a.sort()
print(a)
# 按照每一列进行排序
a.sort(axis=0)
print(a)

array([[12, 11, 10,  9],
       [ 8,  7,  6,  5],
       [ 4,  3,  2,  1]])

array([[ 9, 10, 11, 12],
       [ 5,  6,  7,  8],
       [ 1,  2,  3,  4]])

array([[ 1,  2,  3,  4],
       [ 5,  6,  7,  8],
       [ 9, 10, 11, 12]])

• clip函数 所有小于5的数 以5填充 所有大于9的数以9填充

print(np.clip(a,5,9))

array([[5, 5, 5, 5],
       [5, 6, 7, 8],
       [9, 9, 9, 9]])