benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 212 / I have the following dataframe:

Problem

Answer published by the source. Consult the official source to check your work against its answer.

prompt

Problem:
I have the following dataframe:
  key1  key2
0    a   one
1    a   two
2    b   one
3    b   two
4    a   one
5    c   two

Now, I want to group the dataframe by the key1 and count the column key2 with the value "one" to get this result:
  key1  count
0    a      2
1    b      1
2    c      0

I just get the usual count with:
df.groupby(['key1']).size()

But I don't know how to insert the condition.
I tried things like this:
df.groupby(['key1']).apply(df[df['key2'] == 'one'])

But I can't get any further.  How can I do this?

A:
<code>
import pandas as pd


df = pd.DataFrame({'key1': ['a', 'a', 'b', 'b', 'a', 'c'],
                   'key2': ['one', 'two', 'one', 'two', 'one', 'two']})
</code>
result = ... # put solution in this variable
BEGIN SOLUTION
<code>

library

Pandas

Discussion

Discussion

No discussion posts on this page yet. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

See answer Answer published by the source

Artifacts

Code, notes and reproducible work shared by participants. Files are served from a separate origin.

No artifacts on this page yet. Share reproducible code or notes in a contribution. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

Source and history

upstream_id / Original problem ID
212
task_id
e7118444-1407-5f1c-898f-2af047173aa0
task_key
default--test--212
task_revision_id
1
config / split
default / test

Official source

initial import