创建函数以过滤数据框熊猫

Question

我正在通过实施一些项目来学习Python和Pandas。

我有多个数据框，其中包含有关某些公司的信息。每个公司在每个数据框的一行中。我需要删除一些公司，因此要删除所有数据框中的某些行。

我打算创建一个函数，从所有数据框中删除这些行。像这样：

my_data1 = pd.read_csv(file1.csv)
my_data2 = pd.read_csv(file2.csv)

Company   Inf1 Inf2
c0        1     2
c1        2     4
c2        3     6
c3        4     8
c4        5    10
c5        6    12


list_companies_to_mantain = [c1, c4]

def delete(df):
   df = df[df['Company'].isin(list_companies_to_mantain)

delete(my_data1)
delete(my_data2)


Company   Inf1 Inf2
c1        2     4
c4        5    10

问题是：在函数内部的df中删除了行，但在函数外部没有删除。如果我在函数内部打印（）数据框，它将起作用。但是在功能之外，它不起作用。

这样做的目的是在应用delete（）之后拥有my_data1和my_data2，而没有我不想要的公司。

我觉得答案很明显，但我做不到。如果我不清楚，请告诉我。

谢谢！

Answer 1

您需要return并将其分配回

def delete(df):
   return df[df['Company'].isin(list_companies_to_mantain)

my_data1 = delete(my_data1)