python tips

将字符串列表的元素进行类型转换

str_list = ["1","2","3","4","5"]
#float_list = list(map(float,str_list))
for i in range(str_list):
    str_list[i] = float(str_list[i])

返回list和np.array中的最大值位置

l = [1,2,3,4,5]
index = l.index(max(l))#返回list l中最大值索引

ar_l = np.array(l)
ar_index = np.argmax(ar_l)#返回ar_l中最大值索引

通过一个索引数组来取array中的子集

>>b = np.array([2,3,4,5,6])
>>a = np.array([0.2.4])
>>b[a]
array([2,4,6])

获取ndarray尺寸

#img为一个三维矩阵,获取长，宽，高
l,w,h = img.shape

获取二维ndarray某一行，某一列

#m为一个二维矩阵
m = np.array([[7,5,7,2],
              [2,5,4,3],
              [0,8,5,1],
              [2,4,7,9]])
r = m[2]   #取第二行
c = m[:,2] #取第二列

numpy判断0矩阵，0元素

a2 = np.array([[0,2,3,4],
               [2,3,4,5],
               [3,4,5,6]
               [4,5,6,7]])
>>np.any(a2 == 0) #判断a2中是否含有0元素
>>True
>>np.all(a2 == 0) #判断a2中是否为0矩阵
>>False

提取一维array非零元素的索引

a = np.array([0,1,0,0,1])
b = np.nonzero(a)[0]
>> b
>> [1,4]

numpy将多维数组转为一维

a = np.array([[1,2],
              [3,4]])
b = a.flatten()
>> b
>> [1,2,3,4]

numpy 一维数组的长度

len()

numpy 求矩阵的逆

B = np.linalg.inv(A)