例如我有一个5*5 np.array这样的：
a=[[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]]
如果我想在哪里获得行和列的范围number<=15，该怎么办？
相反，如果我知道行和列的范围（如iinxrange(1,4)和jin）xrange(1,4)，那么如何得到这样的数字：
[[7,8,9],
[12,13,14],
[17,18,19]]
要基于条件获取范围，可以直接应用条件，也可以使用np.where：
>>> 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 < 15
array([[ True, True, True, True, True],
[ True, True, True, True, True],
[ True, True, True, True, False],
[False, False, False, False, False],
[False, False, False, False, False]], dtype=bool)
>>> np.where(a < 15)
(array([0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 2, 2, 2, 2]),
array([0, 1, 2, 3, 4, 0, 1, 2, 3, 4, 0, 1, 2, 3]))
在后一种情况下，返回值是匹配索引的元组。
要实现相反的操作，您可以简单地对数组进行切片：
>>> ar[1:4, 1:4]
array([[ 7, 8, 9],
[12, 13, 14],
[17, 18, 19]])
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句