numpy的简单使用

生成array object

import numpy as np
arr1 = np.arange(10) #1
list1 = range(10)
arr2 = np.array(list1)
list2 = [1, 2, 3, 4]
arr3 = np.array(list2)
print('first:', arr1)
print('second:', arr2)
print('third:', arr3)

array四则运算

arr4 = arr1 + arr2
arr5 = arr1 - arr2
arr6 = np.multiply(arr1, arr2)
arr7 = np.divide(arr1, arr2)
print("addition:", arr4)
print("subtract:", arr5)
print("muptiply:", arr6)
print("divide:", arr7)

随机生成一个矩阵

arr8 = np.random.rand(2, 3)
print("random array:", arr8)

矩阵的各个属性

print("arr8维度ndim：", arr8.ndim)
print("arr8各个维度大小shape：", arr8.shape)
print("arr8类型dtype：", arr8.dtype)

zreos,ones,empty

zeros_arr = np.zeros((2, 3))
ones_arr = np.ones((2, 3))
empty_arr = np.empty((2, 3))
print("zreos_arr:", zeros_arr)
print("ones_arr:", ones_arr)
print("empty_arr:", empty_arr)

identity , astype函数

arr9 = np.identity(5, dtype=np.int32)
print(arr9)
print(arr9.astype(np.float64))

数组条件判断

con_arr = np.array([2.3, 5.3, 6.5, 7.8])
is_true = con_arr >= 5
print(con_arr[is_true])

取值：向上，向下，四舍五入

print(np.ceil(con_arr))
print(np.floor(con_arr))
print(np.rint(con_arr))

聚合函数,reshape函数

print("sum:": np.sum(con_arr))
print("mean:", np.mean(con_arr))
print("max:", np.max(con_arr))
print("min:", np.min(con_arr))
print("方差：", np.var(con_arr))
print("标准差：", np.std(con_arr))
print("reshape:", con_arr.reshape((4, 1)))

累加，累乘

print("累加con_arr:", np.cumsum(con_arr))
print("累乘con_arr:", np.cumprod(con_arr))

where, all , any, unique函数

arr = np.array([5.6, 6.7, 6.7, 5.6, 7.8, 9.8])
print("where:", np.where(arr > 6)) #返回符合条件的下标数组
print("all:", np.all(arr > 6)) #所有元素都大于6才为真
print("any:", np.any(arr > 6)) #只要有一个元素大于6就为真
print("unique", np.unique(arr)) #去唯一的值并且排序