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