gpt4 book ai didi

python - 查找 numpy 3d 数组沿 z 轴的最大索引

转载 作者:太空宇宙 更新时间:2023-11-03 15:06:52 25 4
gpt4 key购买 nike

我有一个 3d numpy 数组。我想找到沿数组 z 轴非零元素的每个 [x,y] 网格点的最大坐标。这将产生一个充满 z 索引值的二维数组。我怎样才能做到这一点?

最佳答案

使用 Numpy 中的 amax 函数,并指定要获取最大值的轴:

ans = numpy.amax(arr_3D, axis=2)

关于python - 查找 numpy 3d 数组沿 z 轴的最大索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44578610/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com