python中内置数据类型list与numpy array都是常会用到的两种数据结构。
二者在访问变量中不连续index时处理方式有所不同。
array
array访问不连续index的方式非常简单,只需要用定义好的索引直接截取array,这得益于array强大的内置方法。
import numpy as np
arr_a = np.random.rand(4)
print(arr_a)
mask = np.ones(len(arr_a),bool)
print(mask)
mask[0],mask[2] = False,False
print(arr_a[mask])
[0.60583009 0.6771267 0.76375809 0.57725833]
[ True True True True]
[False True False True]
[0.6771267 0.57725833]
list
但是list想要做到这件事就会比较麻烦,因为list的内置切片只支持连续的,所以我们需要跑个循环
list_a = list(random(4))
print(list_a)
mask = [True]*4
print(mask)
mask[0],mask[2] = False,False
print(mask)
print([list_a[i] for i,value in enumerate(mask) if value==True])
[0.14190044995211393, 0.14863387410998552, 0.32552951983048084, 0.048596778872093016]
[True, True, True, True]
[False, True, False, True]
[0.14863387410998552, 0.048596778872093016]