由于np.append达不到list中的append的整个列表作为一个整体附在后面的效果,所以这个时候会采用list。但是又需要应用数组的列表索引。
例如:
a=np.array([1,2,3,4,5,6])
index=[True,True,True,False,False,True]
b=a[index]
这时候b的输出为np.array([1,2,3,6])
那么对于列表:
c = [1,2,3,4,5,6]
d=c[index]
这时候d会报错。
我采用的解决办法为:
c = [1,2,3,4,5,6]
d=[c[i] for i in range(len(c)) if index[i]]
同样可以达到把想要的数据挑选出来的效果。