创建过滤数据框架pandas的功能。

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)