我有这个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
是与问题对应的值.