繁体   English   中英

numpy:二维数组,在第一个索引上运行时出错

[英]Numpy: 2d array, error when running over the first index

尝试为100x100 2d数组分配特定值时遇到问题。 我希望它在所有地方都为零,但沿着两个部分除外:第一行从第25行到第75行,最后一列从第25行到第75行。

我尝试使用“:循环”来执行此操作,我的意思是:

my_array[0][25:75] = 1.
my_array[25:75][99] = 1.

但是第二行向我返回错误:

IndexError: index 99 is out of bounds for axis 0 with size 50

好像是先切割长度为50(75-25)的阵列,然后分配长度。

当然,我可以得到想要的东西:

for x in xrange(25,75):
    my_array[x][-1] = 1.

但是,出于好奇,我如何通过:操作完成它?

将两个数组切片放在相同的方括号内。 使问题更小,因此更容易显示:

>>> my_array = np.zeros((10,10))
>>> my_array[0, 2:7] = 1
>>> my_array[2:7, -1] = 1
>>> my_array
array([[ 0.,  0.,  1.,  1.,  1.,  1.,  1.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  1.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  1.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  1.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  1.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  1.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.]])

现在所发生的事情是,你要求这个数组的50排片,然后在[99]正在试图获得的第99排,但只有50。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM