Функция Python для сумм целых чисел

Question

Написал программу python, которая добавила числа от 1 до заданного числа, используя уравнение Гаусса. Он работал на 100 и 10, но когда я делаю 3, он говорит, что сумма равна 4 вместо 6. Уравнение получается в моей голове, я испортил код?

def numberSum(num): 

    nim = int(num) 
    num = (nim/2)*(nim+1) 
    return num  

print numberSum(raw_input()) 

Answer 1

from __future__ import division 

def number_sum(num): 
    """ 
    Return sum of 1 .. num 
    """ 
    return num * (num + 1) // 2 

num = int(raw_input(": ")) 
print(number_sum(num)) 

Что я изменил:

• number_sum не должен нести ответственность за ввод типа литья; если он ожидает номер, вы должны указать его своего рода номера, а не строки.

• , если num является int, то либо num или num + 1 даже, т.е. делением на 2 num * (num + 1) приведет к междунар. Это не обязательно соответствует num; поэтому вы должны разделить продукт на 2, а не num, если вы хотите получить целочисленный результат.

• в Python 2.6+, from __future__ import division делает / всегда возвращают float результат и // всегда возвращают int результат, в соответствии с Python 3.x

Answer 2

Я полагаю, вы работаете с python2, потому что по умолчанию применяется целочисленное деление, как и другие языки. Кроме того, вы можете заметить эту функцию в этом сообщении Integer division in python 2 and python 3.

Чтобы решить проблему, вы можете следовать следующий подход:

def numberSum(num): 
    nim = int(num) 
    return ((nim * 1.0)/2) * (nim + 1)  

print numberSum(raw_input())