如何总结列中包含特定字符串的行中的所有值？

Question

所以我在每个社区都有一个dataframe 的场地类别。 看起来像：

每行中的值代表编号。 特定社区中的每个场所。

我想知道每个街区的餐馆总数。 为此，我知道我必须对列中包含字符串“Restaurant”的行中的值求和。

我试过使用str.contains function 但这总结了真实的情况 - 包含字符串restaurant的列在该行中有多少次值>0 。 但相反，我想要的是，总而言之，总数。 取而代之的是附近的餐馆。

Answer 1

这是一种方法：

df = pd.DataFrame({"restaurant_a": [1,2,3], "shop": [2,3,4], "restaurant_b": [4,5,6]})
df["sum_rest"] = df[[x for x in df.columns if "restaurant" in x]].sum(axis = "columns")
df

结果是：

   restaurant_a  shop  restaurant_b  sum_rest
0             1     2             4         5
1             2     3             5         7
2             3     4             6         9

Answer 2

您可以在此处将pd.Index.str.contains与df.loc一起使用。

df['sum_rest'] = df.loc[:,df.columns.str.contains('Restaurant')].sum(axis=1)

Answer 3

定义一个包含 "Restaurant" 的列列表：

lr = ["Afgan Restaurant", "American Restaurant", "Argentinian Restaurant"]

然后解析结果并将其放入列中：

df["sum_restaurant"] = df.loc[:, columns=lr].apply(lambda row : np.sum(row.to_numpy()))