沿阵列或矩阵轴求和/附加全局，直到满足数值条件

Question

下午。根据标题 - 我有一些问题在数组轴上汇总条目，并将同一行中的相关条目附加到单独的列表中。

背景。我正在使用SVD numpy函数来确定任意数组的奇异值和右奇异向量。我创建了一个numpy数组，其中包含奇异向量和相关值的列。在我正在处理的问题的上下文中 - 我想计算数组中的总和创建一个循环，该循环对行中的条目求和，直到迭代的总和至少为90％（可能只有8个条目中的三个为此，我想创建一个仅包含这些条目的列表及其相关的右侧奇异向量）我现在试图简化我的问题并且只尝试处理一个函数来计算返回关联向量的上述sans。这是我尝试过的快速概述

创建包含奇异值的数组，在本例中为s。 svd函数总共返回3个数组

create the tuples of singular values and associated right singular vectors.

for i in range(len(s)):
singular_pairs=[np.abs(s[i]),v[:,i]]

u,s,v=svd(the matrix)



def variance_threshold(x):
#initialize the sum value, create the empty array of singular values that       will be kept

  x = 0
  kept_eigenvalues=[]
  #condition

  while x < (.9*sum(s)):
    if i <len(s):
      x=s[i]+s[i+1]
      kept_eigenvalues.append(x)
    else:
      break
  return kept_eigenvalues

让我知道这是否可以挽救，或者我是否应该转向嵌套for循环。非常感谢！