将行附加到数据框

我被困在一个简单的任务上。我想创建一个空的 DataFrame 并根据对另一个数据集的查询向其追加行。我已经尝试过这里的答案,但我错过了一些..初学者 Pythoner。任何帮助,将不胜感激。我想取每个状态的前 3 行并将它们添加到一个新的数据框中进行处理。我也尝试附加..


def test():


    #get the list of states

    states_df = census_df.STNAME.unique()

    population_df = pd.DataFrame()


    for st in states_df:

        temp_df = pd.DataFrame(census_df[census_df['STNAME'] == st].nlargest(3,'CENSUS2010POP'))

        pd.concat([temp_df, population_df], ignore_index = True)


    return 1


九州编程
浏览 139回答 2
2回答

慕田峪9158850

我想我知道你在做什么课程,一年前我玩得很开心,坚持下去!我发现连接一堆切片数据帧的最简单/最快的方法是将每个 df 附加到一个列表,然后最后连接该列表。请参阅下面的工作代码(它按照我的意思解释您的意思)。我同意 David 关于排序的建议,更易于使用排序,然后只对前 3 个进行切片。 由于 nlargest() 工作并返回一个我相信的系列而不是数据框,而您想要保留整个数据框结构(所有列) 进行串联。另外为什么你的函数返回1?错别字?我想如果你把它放在一个函数中,你想返回你想要的输出,所以我也改变了它。import pandas as pdimport numpy as np#create fake data random numbersdata = np.random.randint(2,11,(40,3))census_df = pd.DataFrame(index=range(40), columns=['Blah', 'Blah2','CENSUS2010POP'], data=data)#create fake STNAME columncensus_df['STNAME'] = list('aaaabbbbccccddddeeeeffffgggghhhhiiiijjjj')#Function:def test(census_df):    states_list = census_df.STNAME.unique() #changed naming to _list as it's not a df.    list_of_dfs = list() #more efficient to append each df to a list    for st in states_list:        temp_df = census_df[census_df['STNAME']==st]        temp_df = temp_df.sort_values(by=['CENSUS2010POP'], ascending=False).iloc[:3]        list_of_dfs.append(temp_df)    population_df = pd.concat(list_of_dfs,ignore_index=True)    return population_dfpopulation_df = test(census_df)

Cats萌萌

欢迎来到 SO!您的问题是追加还是前三行?对于追加,请尝试df.append函数。它可能看起来像:#get the list of statesstates_df = census_df.STNAME.unique()population_df = pd.DataFrame()for st in states_df:    temp_df = pd.DataFrame(census_df[census_df['STNAME'] == st].nlargest(3,'CENSUS2010POP'))    population_df = population_df.append(temp_df, ignore_index = True) #append the temp df to your main df, ignoring the index对于顶行,您可以使用 df.sort_values(by=['column name'],ascending=False) 然后选择前三行:population_df = population_df.append(temp_df[0:3], ignore_index = True)
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python