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型的,再使用就可以了。

posted @ 2017-02-10 14:18  rainsoul  Views(131945)  Comments(0Edit  收藏  举报