Sum of a column if it contains multiple strings

Question

I want to get the sum of a column based on whether it matches multiple strings. For example,

pd.DataFrame({'key': ['b', 'b d', 'a', 'c t', 'a', 'b p'], 'data1': range(6)})

I want to be able to get the sum for data1 column IF the strings in the key column contain 'b' and 'c'. The sum in this case would be 9.

Thank you for the answer. My question was probably not clear so I have edited it for more clarity. — Salahuddin, Jan 18 '21 at 17:32

score 1 · Accepted Answer · answered Jan 18 '21 at 17:41

1

Try this using regex express and string accessor, .str with contains:

df.loc[df['key'].str.contains('b|c'), 'data1'].sum()

Output:

answered Jan 18 '21 at 17:41

Scott Boston

1 Answers1