如何选择NumPy数组中除索引序列之外的所有元素

11 浏览
0 Comments

如何选择NumPy数组中除索引序列之外的所有元素

假设我有一个长数组和一个索引列表。我该如何选择除这些索引之外的所有元素？我找到了一个解决方案，但它不够优雅：

import numpy as np
x = np.array([0,10,20,30,40,50,60])
exclude = [1, 3, 5]
print(x[list(set(range(len(x))) - set(exclude))])

0