我有一个800x800阵列,我想分析它的外部元素。 我需要一个没有切片元素的新数组[5:-5,5:-5]。 它不一定要返回2d数组,平面数组或列表也可以。 例如:
import numpy
>>> a = numpy.arange(1,10)
array([1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> a.shape = (3,3)
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
我需要丢弃核心元素,例如:
del a[1:2,1:2]
我希望有:
array([1, 2, 3, 4, 6, 7, 8, 9])
我尝试使用numpy.delete()但它似乎一次只适用于一个轴。 我想知道是否有更直接的方法来做到这一点。
答案 0 :(得分:6)
您可以使用布尔数组以您喜欢的方式索引数组。这样,如果您不想更改原始数组中的任何值,则无需这样做。这是一个简单的例子:
>>> import numpy as np
>>> a = np.arange(1,10).reshape(3,3)
>>> b = a.astype(bool)
>>> b[1:2,1:2] = False
>>> b
array([[ True, True, True],
[ True, False, True],
[ True, True, True]], dtype=bool)
>>> a[b]
array([1, 2, 3, 4, 6, 7, 8, 9])
答案 1 :(得分:2)
您可以使用一些占位符值替换中间区域(我使用-12345,实际数据中不会出现的任何内容都可以使用),然后选择不等于该值的所有内容:
>>> import numpy as np
>>> a = np.arange(1,26)
>>> a.shape = (5,5)
>>> a
array([[ 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10],
[11, 12, 13, 14, 15],
[16, 17, 18, 19, 20],
[21, 22, 23, 24, 25]])
>>> a[1:4,1:4] = -12345
>>> a
array([[ 1, 2, 3, 4, 5],
[ 6, -12345, -12345, -12345, 10],
[ 11, -12345, -12345, -12345, 15],
[ 16, -12345, -12345, -12345, 20],
[ 21, 22, 23, 24, 25]])
>>> a[a != -12345]
array([ 1, 2, 3, 4, 5, 6, 10, 11, 15, 16, 20, 21, 22, 23, 24, 25])
如果使用float数组而不是整数数组,则可以使用NaN和isfinite更优雅地执行此操作:
>>> a = np.arange(1,26).astype('float32')
>>> a.shape = (5,5)
>>> a[1:4,1:4] = np.nan
>>> a
array([[ 1., 2., 3., 4., 5.],
[ 6., nan, nan, nan, 10.],
[ 11., nan, nan, nan, 15.],
[ 16., nan, nan, nan, 20.],
[ 21., 22., 23., 24., 25.]], dtype=float32)
>>> a[np.isfinite(a)]
array([ 1., 2., 3., 4., 5., 6., 10., 11., 15., 16., 20.,
21., 22., 23., 24., 25.], dtype=float32)