python - 如何修改函数中的 pandas DataFrame 以便调用者看到更改？

Question

我发现自己对各种 [pandas][1] DataFrames 重复执行任务，所以我创建了一个函数来进行处理。如何修改函数 process_df(df) 中的 df 以便调用者看到所有更改(无需分配返回值)？

代码的简化版本:

def process_df(df):
    df.columns = map(str.lower, df.columns)

df = pd.DataFrame({'A': [1], 'B': [2]})
process_df(df)
print df

   A  B 
0  1  2

编辑新代码:

def process_df(df):
    df = df.loc[:, 'A']

df = pd.DataFrame({'A': [1], 'B': [2]})
process_df(df)
print df

   A  B 
0  1  2

Answer 1

使用 ix、loc、iloc 等索引 DataFrame 返回底层数据的 View (这是一个读操作)。为了修改框架的内容，您需要使用就地转换。例如，

def process_df(df):
    # drop all columns except for A
    df.drop(df.columns[df.columns != 'A'], axis=1, inplace=True)

df = DataFrame({'A':[1,2,3], 'B':[1,2,3]})
process_df(df)

要改变列的顺序，你可以这样做:

def process_df(df):
    # swap A and B
    df.columns = ['B', 'A']
    df[['B', 'A']] = df[['A', 'B']]