lyh*_*198 3 python dataframe pandas
所以我有每个社区的场地类别的数据框。看起来像:
每行中的值代表编号。特定社区中每个场馆的信息。
我想找出每个街区的餐馆总数。为此,我知道我必须对列包含字符串“Restaurant”的行中的值进行求和。
我尝试过使用str.contains函数,但这总结了真实情况 - 包含字符串的列在该行中restaurant具有值的次数>0。但相反,我想要的是,总而言之,总数没有。而是附近的餐馆。
您可以pd.Index.str.contains 在df.loc这里使用。
df['sum_rest'] = df.loc[:,df.columns.str.contains('Restaurant')].sum(axis=1)
Run Code Online (Sandbox Code Playgroud)