1. 程式人生 > 程式設計 >python之array賦值技巧分享

python之array賦值技巧分享

首先上一段程式:

import numpy as np

list_a = list(range(10))
print("list_a: {}".format(list_a))

array_a = np.array(list_a)
print("array_a: {}".format(array_a))

list_b = [True,False,True,False]
print("list_b: {}".format(list_b))

array_b = np.array(list_b)
print("array_b: {}".format(array_b))

mask = np.zeros(10,dtype=np.bool)
print("mask: {}".format(mask))

下面是這段程式的輸出:

list_a: [0,1,2,3,4,5,6,7,8,9]
array_a: [0 1 2 3 4 5 6 7 8 9]
list_b: [True,False]
array_b: [ True False True True False False True False True False]
mask: [False False False False False False False False False False]

從輸出可以看出list和array的不同之處是list是以逗號作為分隔符,而array是以空格作為分隔符。還有numpy中的zeros生成的是array。其實,list和array還有很多的不同之處,例如:

list是Python中的普通列表物件,支援append和attend操作,沒有shape屬性;array是numpy資料庫中的物件,不支援append和attend操作,具有shape屬性。

一個list中可以存放不同型別的資料,如int、float、str或者布林型;而array中只能存放相同型別的資料。

list不支援乘法操作;array支援乘法操作。

而對於賦值操作,操作物件只能是array,而索引可以是array也可以是list:

d = array_a[list_b]
print("array_a[list_b]: {}".format(d))

e = array_a[array_b]
print("array_a[array_b]: {}".format(e))

mask[list_b] = 1
print("mask: {}".format(mask))

輸出如下:

array_a[list_b]: [0 2 3 6 8]
array_a[array_b]: [0 2 3 6 8]
mask: [ True False True True False False True False True False]

此番操作可以將array_a中對應b陣列或列表True位上的元素組成一個新的陣列。

如果對列表進行如此賦值操作,即把程式中的array_a換成list_a,則會報錯:

d = list_a[list_b]
TypeError: list indices must be integers or slices,not list

e = list_a[array_b]
TypeError: only integer scalar arrays can be converted to a scalar index

以上這篇python之array賦值技巧分享就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。