Python Pandas根据条件添加列值

Question

我有一个具有下列值的DataFrame (df)：

                  Title
fintech_countries          
US                 60
UK                 54
India              28
Australia          25
Germany            13
Singapore          11
Canada             10

我想添加所有值小于25的国家/地区，并将它们与其总和(34)一起显示为“其他”。

我已经通过以下代码创建了国家/地区的列名：

df1 = df.rename_axis('fintech_countries').rename_axis("countries", axis="columns" , inplace=True)


countries         Title
fintech_countries          
US                 60
UK                 54
India              28
Australia          25
Germany            13
Singapore          11
Canada             10

现在，我已经基于StackOverflow上的另一个查询尝试了以下代码：

df1.loc[df1['Title'] < 25, "countries"].sum()

但是我得到了以下错误：

KeyError: 'the label [countries] is not in the [columns]'

有人能帮帮忙吗？我需要如下的最终输出：

countries         Title
fintech_countries          
US                 60
UK                 54
India              28
Australia          25
Others             34

提亚

Answer 1

使用loc实现setting with enlargement和按boolean indexing过滤的解决方案

mask = df['Title'] < 25
print (mask)
fintech_countries
US           False
UK           False
India        False
Australia    False
Germany       True
Singapore     True
Canada        True
Name: Title, dtype: bool

df1 = df[~mask].copy()
df1.loc['Others', 'Title'] = df.loc[mask, 'Title'].sum()
df1.Title = df1.Title.astype(int)
print (df1)
countries          Title
fintech_countries       
US                    60
UK                    54
India                 28
Australia             25
Others                34