gpt4 book ai didi

python - 在 Python 的 scipy/numpy 中有效地找到非零区间?

转载 作者:太空狗 更新时间:2023-10-30 02:14:18 25 4
gpt4 key购买 nike

假设我有一个 python 列表或 python 一维数组(以 numpy 表示)。假设有一段连续的元素,我怎样才能找到这个列表或数组中非零段的开始和结束坐标(即索引)?例如,

a = [0, 0, 0, 0, 1, 2, 3, 4]

nonzero_coords(a) 应该返回 [4, 7]。对于:

b = [1, 2, 3, 4, 0, 0]

nonzero_coords(b) 应该返回 [0, 2]。

谢谢。

最佳答案

假设有一段连续的非零元素......

x = nonzero(a)[0]
result = [x[0], x[-1]]

关于python - 在 Python 的 scipy/numpy 中有效地找到非零区间?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/2619413/

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