如何看待NumPy数组中的第一个非零数字?

Question

我有一个立方体a = numpy.array(n,n,n)。大小会有所不同。

我想从顶部看，所以在列上，有nxn的，并返回一个数组(n,n)，其中包含你可以看到的第一个非零数字。

这就像往下看数组，想象0是洞。

我可以使用sum(a, axis=0)，但这也会将第一个零以下的任何数字相加。我只需要0后面的第一个数字(如果列中有n个0则为0)

我希望这足够清楚，以获得一些建议:)

欢呼,保罗

Answer 1

你可以用np.where(a==0)找到你搜索的元素，如果有多个，你可以用np.where(a==0)[0][0]找到第一个索引所以要得到零后的第一个数字你应该用a[np.where(a==0)[0][0]+1]