排除元素我想要得到C數組的結果,但我不知道如何:從numpy的陣列
import numpy as np
a = xrange(10)
b = np.array([3,2,1,9])
使c爲不在B的元素:
c = np.array([0,4,5,6,7,8])
排除元素我想要得到C數組的結果,但我不知道如何:從numpy的陣列
import numpy as np
a = xrange(10)
b = np.array([3,2,1,9])
使c爲不在B的元素:
c = np.array([0,4,5,6,7,8])
也許更直接的解決方案是:
import numpy as np
a = xrange(10)
b = np.array([3,2,1,9])
c = np.setdiff1d(a,b)
導致:
In [7]: c
Out[7]: array([0, 4, 5, 6, 7, 8])
你可以找到所有的說明文檔中的numpy的陣列集合類業務: http://docs.scipy.org/doc/numpy/reference/routines.set.html
import numpy as np
a = np.arange(10)
b = np.array([3,2,1,9])
np.array(sorted(set(a) - set(b)))
# array([0, 4, 5, 6, 7, 8])
UPDATE:與a = xrange(10)
一起工作。
什麼是'np.xrange'?你的意思是'np.arange'? – eumiro 2011-12-31 13:05:35