数组中的元素来自总和

Question

我是python中的新手，我有一个我无法弄清楚的简单问题。

我有一个数组让我们说A = [1 2 3 4 5] 如果我们总结数组的元素，它将给出15.例如，我想要前3个元素的总和，所以我将有6个。

我的问题是我想要一个新数组，它会从总和中给出前3个元素（让我们调用该变量my_sum）。那就是：B=[1 2 3]

我陷入了我的代码：

my_sum = [0]

for num in A:

    if my_sum !=6:

        my_sum = my_sum + num

之后，我不知道该怎么做。

Answer 1

喜欢那个？！

import numpy

a = numpy.array([1, 2, 3, 4, 5, 6])
b = numpy.array([])

sum = 0
for num in a:
    if sum != 6:
       sum += num
       b = numpy.append(b, num)
    else:
       break

print b.astype(int) # [1 2 3]