【Python】数组的条件逻辑&统计运算&元素排序

一.条件逻辑

import numpy as np
arr_x = np.array([1, 5, 7])
arr_y = np.array([2, 6, 8])
arr_con = np.array([True, False, True])
result = np.where(arr_con,arr_x,arr_y)
result

二.统计运算

arr = np.arange(10)
arr

arr.sum()         #求和

arr.mean()        #求平均值

arr.min()         #求最小值

arr.max()         #求最大值

arr.argmin()      #求最小值的索引

arr.argmax()      #求最大值的索引

arr.cumsum()      #计算元素的累加和

arr.cumprod()    #计算元素的累乘积

三.数组元素排序

arr = np.array([[6, 2, 7], 
                [3, 6, 2], 
                [4, 3, 2]])
arr.sort()
arr

arr.sort(0)        #sort()方法会修改数组本身
arr

检索数组元素是否满足条件

arr = np.array([[1, -2, -7], [-3, 6, 2], [-4, 3, 2]])
np.any(arr > 0)              #arr所有的元素是否有一个大于0

np.all(arr > 0)              #arr所有的元素是否都大于0

查找数组的唯一元素

arr = np.array([12, 11, 34, 23, 12, 8, 11])
np.unique(arr)

判断元素是否在其他数组中

arr = np.array([12, 11, 34, 23, 12, 8, 11])
arr2 = np.array([11,12])
np.in1d(arr,arr2)