Question

我编写的代码确定哪些小于或等于1 000 000的自然数是完美的数字到数字不变量（它们可以写成数字的总和，每个数字都增加到等于数字的幂;请参阅https://en.wikipedia.org/wiki/Perfect_digit-to-digit_invariant）并以此形式表达。

def f(n):
    y = str(n)
    l = len(y)
    list = []
    for i in range(0,l):
        list.append(int(y[i]))
    list2 = []
    for i in list:
        list2.append(i**i)
    return sum(list2)

N = 10**6
n = np.linspace(1,N,N,dtype=int)
list = []
for i in n:
    if i == f(i):
        list.append(i)

list = np.array(list)

list2 = []
for i in list:
    list2.append(str(i))

for i in list2:
    for j in range(0,len(i)-1):
        print(i[j],'^',i[j],'+')
    print(i[-1],'^',i[-1],'=',int(i))
    print('----------')

运行此代码的输出是：

1 ^ 1 = 1
----------
3 ^ 3 +
4 ^ 4 +
3 ^ 3 +
5 ^ 5 = 3435
----------

代码给出了正确答案，但我希望表达式3 ^ 3 + 4 ^ 4 + 3 ^ 3 + 5 ^ 5 = 3435出现在一行上。有没有办法做到这一点？

Answer 1

默认情况下，

print()语句也会打印换行符。要禁止该行为并继续在同一行上打印，请添加end=''参数，如下所示：

for i in list2:
    for j in range(0, len(i)-1):
        print(i[j],'^',i[j],'+', end='')
    print(i[-1],'^',i[-1],'=',int(i))
    print('----------')

或者，更多的是，您可以通过join()字符串组件替换内部循环并按如下方式格式化输出：

for i in list2:
    print(' + '.join(['{0:}^{0:}'.format(x) for x in i]) + ' = {}'.format(i))
    print('----------')

Answer 2

尝试打印

中的end=" "选项

for j in range(0,len(i)-1):
    print(i[j],'^',i[j],'+', end=" ")

用于数字到数字不变量的Python代码

2 个答案: