在另一个python文件中导入函数后如何以某种方式执行函数?

时间:2016-10-05 02:41:35

标签: python python-2.7

这是一个函数,它返回从0到9的随机数长度列表。

import random

def makeList():
    y=[]
    n=int(raw_input("Enter number"))

    for x in range(0,n):
        x=random.randint(0,9)
        y.append(x)
    print y

def main():
    makeList()
if __name__ == '__main__':
    main()

现在我正在创建另一个名为main.py的python文件并导入此函数 这是我在main.py中编写的代码。该文件的名称是myFunc。

from myFuncs import makeList
makeList()

现在,     我想创建一个列表(比如10个随机数。)我应该在main.py文件中写什么来实现这个目标?

1 个答案:

答案 0 :(得分:0)

在第一个文件中将print y更改为return y。 (以及print(makeList())下的main()

在新文件new_list = makeList()

然后new_list就是那个生成的列表。

编辑:更多pythonic版本

文件1

import random

def make_list(length=10):
    return [random.randint(0,9) for _ in range(length)]

def main():
    amount = int(raw_input("Enter number"))
    print(make_list(amount))

if __name__ == '__main__':
    main()

文件2

from myFuncs import make_list

new_list = make_list()
print(new_list) # will be default 10

<强> -

如果你想将输入保持在同一个功能中,可以采用以下方法:

def make_list(length=0):
    if not length:
        length= int(raw_input("Enter number"))    
    return [random.randint(0,9) for _ in range(length)]

def main():
    print(make_list())

然后在第二个文件中指定make_list(10)