list和array 访问不连续index

python中内置数据类型list与numpy array都是常会用到的两种数据结构。 二者在访问变量中不连续index时处理方式有所不同。

array

array访问不连续index的方式非常简单，只需要用定义好的索引直接截取array，这得益于array强大的内置方法。

代码语言：javascript复制

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])

代码语言：javascript复制

[0.60583009 0.6771267  0.76375809 0.57725833]
[ True  True  True  True]
[False  True False  True]
[0.6771267  0.57725833]

list

但是list想要做到这件事就会比较麻烦，因为list的内置切片只支持连续的，所以我们需要跑个循环

代码语言：javascript复制

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])

代码语言：javascript复制

[0.14190044995211393, 0.14863387410998552, 0.32552951983048084, 0.048596778872093016]
[True, True, True, True]
[False, True, False, True]
[0.14863387410998552, 0.048596778872093016]

python numpy 数据结构 list 数据类型

0 人点赞