在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