在python中定义一个主要函数，该函数可以在python中运行其他函数

Question

我有一个具有功能的file1.py文件。

def func1(x):
    ....
    ....
    ....
    return x

我也有一个数据框df 。

我正在使用.apply function运行此.apply function

df['col1'].apply(func1)

但是为此，我必须打开.py文件并运行它。

我想从命令行运行.py文件，以便直接进入df['col1'].apply(func1)

所以我想定义一个主函数，它将运行我的df['col1'].apply(func1)

我对python的了解非常有限，有什么办法吗？

Answer 1

您可以将其放在file1.py ：

import pandas as pd

df = pd.DataFrame([1, 2, 3, 4, 5, 6, 7])

def func1(x):
    x = x * 2
    return x

print df.apply(func1)

然后从命令行运行：

python file1.py