从给定索引列表的列表中删除多个条目

6ljaweal  于 2021-09-08  发布在  Java
关注(0)|答案(1)|浏览(308)

我有一个索引列表和一个数据列表。索引列表说明哪些元素应该从数据列表中删除。我希望有效地使用索引列表,即没有循环。有没有更快的方法删除这些元素?

muk1a3rh

muk1a3rh1#

假设你使用numpy, np.delete 完全符合您的要求:

>>> a = np.array([1, 4, 9, 16, 25, 36])
>>> np.delete(a, [1, 2, 5])
array([ 1, 16, 25])

相关问题