基于创建从列表蟒蛇列表中，如果条件

Question

我想创建一个new_list这将只包含满足条件的我old_list的项目，在标签的数组索引3.我想是这样的：

new_list = [x for x in old_list if idx[x] == 3] 
IndexError: arrays used as indices must be of integer (or boolean) type 

但我得到以下错误，因为idx是一个数组。我怎么解决这个问题？

编辑：IDX是相等尺寸的阵列与我的含有标签他们原始数据。所以基本上我想创建一个新的列表，将只包含我的原始列表中的项目，例如有标签3.

我想要做这样的事情： cluster_a = [old_list [x] for x in idx如果x == 3]

澄清：我的旧列表是包含3D阵列列表以及IDX是含有一个标签为我的列表中的每个3D阵列作为我上述相同的大小数组。我正在尽力解释这个问题。如果需要什么，请告诉我。

Answer 1

的问题不在于idx是一个列表，但可能是x是一个数组 - old_list必须包含一个列表作为元素。您需要引用索引，而不是项目本身：

[old_list[x] for x in range(len(old_list)) if idx[x] == 3] 

这里有一个小例子：

>>> old_list = [4,5,6] 
>>> idx = [3,2,3] 
>>> [old_list[x] for x in range(len(old_list)) if idx[x] == 3] 
[4, 6] 

Answer 2

这个是什么？ ：

new_list = [x for x in old_list if idx.index(x) == 3 ]