categories = ["护肤&头发", "彩妆&香水"]

df_skinhair = pd.DataFrame() df_makeupfrag = pd.DataFrame()

for category in categories: df_category = filtered_df[filtered_df[1] == category].reset_index(drop=True) df_category = df_category.dropna(axis=1, how='all') df_category.columns = df_category.iloc[0] df_category = df_category.drop([0, 1]).reset_index(drop=True) df_category = df_category.drop(df_category.columns[[0]], axis = 1)

if category == "护肤&头发":
    df_skinhair = df_category
elif category == "彩妆&香水":
    df_makeupfrag = df_category

df_skinhair df_makeupfra

用for循环list等做法整合以下两段代码让它们更简洁df_skinhair = filtered_dffiltered_df1 == 护肤&头发reset_indexdrop=True #筛选护肤&头发类df_skinhair = df_skinhairdropnaaxis=1 how=alldf_skinhaircolumns = df_skinhairiloc0df_skinhair = d

原文地址: https://www.cveoy.top/t/topic/imS0 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录