基於創建從列表蟒蛇列表中，如果條件

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 ]