是否有更多的pythonic方法来编写只更新变量的while循环？

发布时间：2020-12-16 23:37:33 所属栏目：Python 来源：网络整理

导读：我有这个while循环,我想知道它们是否是一种更加 pythonic的方式来编写它： k = 1while np.sum(s[0:k]) / s_sum retained_variance: k += 1 s是一个numpy向量.谢谢！解决方法可能不是最有效的解决方案,但如果需要搜索大多数阵列,则速度很快： import numpy

我有这个while循环,我想知道它们是否是一种更加 pythonic的方式来编写它：

k = 1
while np.sum(s[0:k]) / s_sum < retained_variance:
    k += 1

s是一个numpy向量.谢谢！

解决方法

可能不是最有效的解决方案,但如果需要搜索大多数阵列,则速度很快：

import numpy as np

ss = np.cumsum(s)  # array with cumulative sum
k = ss.searchsorted(retained_variance*s_sum) # exploit that ss is monotonically increasing

编辑：西蒙指出

k = np.cumsum(s).searchsorted(retained_variance*s_sum) + 1

是与问题对应的值.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!