gpt4 book ai didi

python - 删除numpy数组的空维度

转载 作者:行者123 更新时间:2023-12-04 00:03:38 26 4
gpt4 key购买 nike

我有一个形状(X,Y,Z)的numpy数组。我想检查每个 Z 维度并快速删除非零维度。

详细解释:

我想检查 array[:,:,0] 是否有任何条目非零。

如果是,忽略并检查数组[:,:,1]。

否则,如果否,则删除维度数组[:,:,0]

最佳答案

也不是 100% 确定你想要什么,但我认为你想要

np.squeeze(array, axis=2)

https://docs.scipy.org/doc/numpy-1.13.0/reference/generated/numpy.squeeze.html

关于python - 删除numpy数组的空维度,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53980994/

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