有没有办法使用 for 循环使它更简洁？

Question

def rydbergEquation(finalEnergyLevel, initialEnergyLevel):
  return ((1.0974 * 10**7) * (1 / (finalEnergyLevel**2)) - (1 / (initialEnergyLevel**2)))**-1

print("Energy Level (2 to 1):     ", rydbergEquation(1, 2))
print("Energy Level (3 to 1):     ", rydbergEquation(1, 3))
print("Energy Level (4 to 1):     ", rydbergEquation(1, 4))
print("Energy Level (5 to 1):     ", rydbergEquation(1, 5))
print("Energy Level (6 to 1):     ", rydbergEquation(1, 6))
print("Energy Level (7 to 1):     ", rydbergEquation(1, 7))
print("Energy Level (3 to 2):     ", rydbergEquation(2, 3))
print("Energy Level (4 to 2):     ", rydbergEquation(2, 4))
print("Energy Level (5 to 2):     ", rydbergEquation(2, 5))
print("Energy Level (6 to 2):     ", rydbergEquation(2, 6))
print("Energy Level (7 to 2):     ", rydbergEquation(2, 7))
print("Energy Level (4 to 3):     ", rydbergEquation(3, 4))
print("Energy Level (5 to 3):     ", rydbergEquation(3, 5))
print("Energy Level (6 to 3):     ", rydbergEquation(3, 6))
print("Energy Level (7 to 3):     ", rydbergEquation(3, 7))
print("Energy Level (5 to 4):     ", rydbergEquation(4, 5))
print("Energy Level (6 to 4):     ", rydbergEquation(4, 6))
print("Energy Level (7 to 4):     ", rydbergEquation(4, 7))
print("Energy Level (6 to 5):     ", rydbergEquation(5, 6))
print("Energy Level (7 to 5):     ", rydbergEquation(5, 7))
print("Energy Level (7 to 6):     ", rydbergEquation(6, 7))

我可以删除 21 个 print 语句并创建一个 for 循环来输出给定值吗？

我似乎找不到一种方法来循环函数以获得不同能级变化的输出。我假设您可以遍历该范围并输出给定值，但我无法找到正确的方法来做到这一点而不会使输出看起来不正确。

通过使用某种双重 for 循环，我可以看到它在工作。

解决方案： itertools 有一个很棒的组合方法，我最终使用了它。

Answer 1

您可以使用 itertools.combinations 来达到这个目的：

import itertools as it

for i, j in it.combinations(range(1, 8), r=2):
    print(f"Energy Level ({j} to {i}):     ", rydbergEquation(i, j))

Answer 2

Itertools 可能更好，但对于小情况，您可能可以使用双 for 循环：

for i in range(6):
    for j in range(i+1, 7):
        print(f"Energy Level ({i+1} to {j+1}):     ", rydbergEquation(i+1, j+1))

用这个替换你所有的打印语句。

有没有办法使用 for 循环使它更简洁？

Is there a way to use a for loop to make this more concise?

python

loops

for-loop

repeat

python-3.x