python中的list以及list与array相互转换
python中的list是一种有序集合,可以随时增删元素;
1 # -*- coding: utf-8 -*- 2 3 frameID = 1 4 frameID_list = [] 5 frameID_list.append(frameID) 6 print (frameID_list) 7 frameID = 2 8 frameID_list.append(frameID) 9 print (frameID_list) 10 name = 'SRX' 11 frameID_list.append(name) 12 print (frameID_list) 13 print (len(frameID_list)) #打印长度 14 print (frameID_list[2]) #使用[]进行索引 15 print (frameID_list[-1]) 16 frameID_list.insert(1, 'WJC') #在指定位置插入元素 17 print (frameID_list) 18 frameID_list.pop() #将最后一位的元素删除 19 print (frameID_list) 20 frameID_list.pop(0) #删除指定位置的元素 21 print (frameID_list)
下面说一下list与numpy中的array相互转换。
首先list中的元素属性应该一致才能转换,如下所示,其中frame_ID_list中包含的元素是str型记录的帧号,首先利用np.array将其转成int型的array,然后将转换之后的数组每个元素都做了一个数学运算,由于运算之后的结果是float,因此在第二行又将其强转成了int型,最后再将np的数组转回list,整个操作相当于完成了对list(其中包含的是str类型的元素)中的每个元素进行数学运算,然后再转回list的过程。
1 frame_ID_list_np = np.array(frame_ID_list, dtype = int) 2 frame_ID_list_np = np.array(frame_ID_list_np / 1000 * fps, dtype = int) 3 frame_ID_list = frame_ID_list_np.tolist()
另外,还有一个需要注意的地方,就是类型转换之后再判断两个变量是否相等时,例如:
frame_ID = '1000'
ID_count = 1000
1 frame_ID = '1000' 2 ID_count = 1000 3 if frame_ID == ID_count: 4 ...
这时这个if是不会成立的,这么直接写出来就很容易看到错误所在,但是当frame_ID与ID_count是从别的变量获得的,就没有那么清晰的看出两者的类型了,所以在编程的时候一定要注意这些小细节;
类似的还有存储文件的时候,比如:
1 str_name_FID = str(frame_ID_list_np_save[count_save_frame_np]) 2 cv2.imwrite('image/' + video_name_save + '_' + str_name_FID + '.jpg',frame) #save the image 3 count_save_frame_np = count_save_frame_np + 1
在存储图片时,图片的名字一定都要是str类型相互串在一起的,count_save_frame_np是int型的变量,如果直接在程序中使用该变量作为图片名的一部分,是不会存下来任何文件的,所以第一行的强转是必须的,先将其转成str型的,再使用就可以了。