|
|
相关文章推荐
|
酒量大的哑铃 · python:将numpy数组数据类型从in ...· 2 周前 · |
|
|
有腹肌的大白菜 · MethodHandles.DropArgu ...· 1 年前 · |
|
|
瘦瘦的鸡蛋面 · map 类 | Microsoft Learn· 1 年前 · |
|
|
会搭讪的领结 · javascript - While ...· 1 年前 · |
|
|
冷静的楼房 · 快速入门：Azure Blob 存储库 - ...· 1 年前 · |
|
|
奔放的梨子 · Windows 窗体设计器入门 - EF ...· 1 年前 · |
腾讯云
问
如何删除numpy数组中的特定元素
如何删除numpy数组中的特定元素
问
如何删除numpy数组中的特定元素
EN
EN
Stack Overflow用户
提问于
2012-06-12 19:54:38
回答 8
查看 654.6K
关注 0
票数 283
如何从numpy数组中删除一些特定的元素？假设我有
代码语言：
javascript
复制
import numpy as np
a = np.array([1,2,3,4,5,6,7,8,9])
然后我想从
a
中删除
3,4,7
。我只知道值的索引(
index=[2,3,6]
)。
EN
回答 8
Stack Overflow用户
发布于 2016-04-07 03:33:33
有一个numpy内置函数可以帮助你做到这一点。
代码语言：
javascript
复制
import numpy as np
>>> a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> b = np.array([3,4,7])
>>> c = np.setdiff1d(a,b)
array([1, 2, 5, 6, 8, 9])
票数 93
EN
Stack Overflow用户
发布于 2019-04-26 17:43:13
要按值删除：
代码语言：
javascript
复制
modified_array = np.delete(original_array, np.where(original_array == value_to_delete))
票数 30
EN
Stack Overflow用户
发布于 2012-06-12 20:06:15
不是一个麻木的人，我尝试了一下：
代码语言：
javascript
复制
>>> import numpy as np
>>> import itertools
>>> a = np.array([1,2,3,4,5,6,7,8,9])
>>> index=[2,3,6]
>>> a = np.array(list(itertools.compress(a, [i not in index for i in range(len(a))])))
array([1, 2, 5, 6, 8, 9])
根据我的测试，它的性能优于
numpy.delete()
。我不知道为什么会出现这种情况，可能是因为初始数组太小了？
代码语言：
javascript
复制
python -m timeit -s "import numpy as np" -s "import itertools" -s "a = np.array([1,2,3,4,5,6,7,8,9])" -s "index=[2,3,6]" "a = np.array(list(itertools.compress(a, [i not in index for i in range(len(a))])))"
100000 loops, best of 3: 12.9 usec per loop
python -m timeit -s "import numpy as np" -s "a = np.array([1,2,3,4,5,6,7,8,9])" -s "index=[2,3,6]" "np.delete(a, index)"
推荐文章
|
|
瘦瘦的鸡蛋面 · map 类 | Microsoft Learn 1 年前 |