return 剥离数据帧的函数

Question

我有一个来自 CSV 文件的数据框：

import pandas as pd

filename = 'mike.csv'
main_df = pd.read_csv(filename)

我需要一个函数来从空格中剥离所有字符串列（也有数字列）的内容，然后 return 这样剥离的数据框。在下面的函数中，剥离似乎工作正常，但我不知道如何 return 剥离的数据帧：

def strip_whitespace(dataframe):
    dataframe_strings = dataframe.select_dtypes(['object'])
    dataframe[dataframe_strings.columns] = dataframe_strings.apply(lambda x: x.str.strip())
    return # how to return a stripped dataframe here?

完整代码：

import pandas as pd

filename = 'mike.csv'
main_df = pd.read_csv(filename)

def strip_whitespace(dataframe):
    dataframe_strings = dataframe.select_dtypes(['object'])
    dataframe[dataframe_strings.columns] = dataframe_strings.apply(lambda x: x.str.strip())
    return stripped_dataframe # ?

stripped_main_df = strip_whitespace(main_df) # should be stripped df

Answer 1

我认为 read_csv 中需要参数 skipinitialspace=True:

main_df = pd.read_csv(filename, skipinitialspace=True)

然后不需要剥离列。

但如果需要使用您的功能：

return dataframe

return 剥离数据帧的函数

Function to return stripped dataframe

python

pandas

function

strip

removing-whitespace