2

我正在为 Coursera 上的数据科学简介课程解决一个问题，并且我正在努力将数据添加到数据框中的列中。

这是我正在使用的数据集：

    SUMLEV  REGION  DIVISION    STATE   COUNTY  STNAME  CTYNAME     
1   50      3       6           1       1       Alabama Autauga County  
2   50      3       6           1       3       Alabama Baldwin County  
3   50      3       6           1       5       Alabama Barbour County  
4   50      3       6           1       7       Alabama Bibb County

我要做的是插入一个名为 TotalCounties 的列，其中包含按州划分的县总数作为最后一列。我在 SQL 中做过类似的事情，但在 Python 中似乎不太一样。

我已经尝试了下面的代码，但该列最终显示为 NaN 而不是我想要的数字。

   counties_only_df = census_df[census_df['SUMLEV'] == 50]
   x = counties_only_df.groupby('STNAME').count()['SUMLEV']
   counties_only_df['Total Counties'] = x

我想在新创建的列中显示一个数字而不是 NaN。

4

1 回答 1

0

尝试：

df['Total Counties'] = df.groupby('STNAME')['STNAME'].transform('count')

更改“df”为您的 DataFrame 名称

于 2019-07-17T21:38:24.253 回答