1. 程式人生 > >python中的list以及list與array相互轉換

python中的list以及list與array相互轉換

轉載出處:http://www.cnblogs.com/rainsoul/p/6386174.html

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