如何修复函数,以便所得数据框仅具有子集的列?

问题描述 投票:-1回答:1

试图对数据框进行子集化以从给定的数据框中删除/删除某些列请帮助我修复此功能

dataframe = pd.DataFrame({
    "X": range(10),
    "Y": range(10,20),
    "Z": range(5,15)
})

def subset_dataframe(dataframe, drop_cols):
    subset_df = data.drop([drop_cols], inplace=True)
    return subset_df


subsetted_df = subset_dataframe(dataframe, drop_cols=["x", "z"]) 
python pandas dataframe
1个回答
0
投票
dataframe = pd.DataFrame({ "X": range(10), "Y": range(10,20), "Z": range(5,15) })

def subset_dataframe(dataframe, drop_cols): subset_df = data.drop(columns = [drop_cols], inplace=True) return subset_df

subsetted_df = subset_dataframe(dataframe, drop_cols=["x", "z"]) 

也在data.drop($)中也添加列

编辑1:

最终功能:

dataframe = pd.DataFrame({ "X": range(10), "Y": range(10,20), "Z": range(5,15) })

def subset_dataframe(dataframe, drop_cols): 
    subset_df = dataframe.drop(columns = drop_cols) 
    return subset_df

subsetted_df = subset_dataframe(dataframe, drop_cols=["X", "Z"])

编辑2:

如果添加inplace = True,则对象以Nonetype返回,并且没有任何内容存储在subset_df中。还要检查语法。编辑1中的最终功能有效。希望这会有所帮助:)

© www.soinside.com 2019 - 2024. All rights reserved.