导入功能时，附加了其他说明

Question

我有一个包含函数的.py文件：

import random
def random_m(a):
b = [[0 for _ in range(a)] for _ in range(a)]
for i in range(a):
    for j in range(a):
        b[i][j] = random.randint(100,999)
    print('{}'.format(b[i],[j]))
n = int(input("Enter dimension of matrix: "))
random_m(n)

现在，我想在另一个.py文件中使用此功能。我写：

from file1.py import random_m 
random_m(10)

执行文件时，它一直要求我输入矩阵的维度。为什么？该指令不包括在功能的身体中。我已经插入了论点" 10"。

感谢您的帮助！

Answer 1

def random_m(a):
  b = [[0 for _ in range(a)] for _ in range(a)]
  for i in range(a):
    for j in range(a):
      b[i][j] = random.randint(100,999)
    print('{}'.format(b[i],[j]))
if __name__ == '__main__':
  n = int(input("Enter dimension of matrix: "))
  print(random_m(n))

这将仅在未导入模块时执行输入函数（以及if __name__ == '__main__':下的所有内容）（例如，当您双击Windows中时）