如何编写一个计算整数平方并将其添加到字典的函数?

时间:2019-04-25 17:50:40

标签: python-3.x

编写一个函数,该函数采用小于15的一个整数并确定从1到该数字(包括)的所有整数的平方。然后将它们放入字典,其中的键是整数,值是平方。如果输入的数字大于15并且不进行任何计算,则该函数必须显示警告。

这是我到目前为止所拥有的。我不知道该如何解决。请帮助

a=1
squared=a**2
first_dict={'a':'a**2'}

while a<15:
    print(a**2)
    a=a+1
if a>15:
    print("Warning: Input number is greater than 15")
    print("0")

1 个答案:

答案 0 :(得分:0)

首先,您没有定义函数,您需要这样做:

def get_squares(n):

    if n > 15:
        print("n greater than 15")
        return None

    result = {}

    for i in range(1, n+1):
        result[i] = i ** 2

    return result



r1 = get_squares(4)
print(r1)

get_squares(16)

for循环可以替换为列表理解,例如:

return {i: i**2 for i in range(1, n+1)}

在控制台中执行的操作会产生:

{1: 1, 2: 4, 3: 9, 4: 16}
n greater than 15