{"kind":"task","effective_mode":"full","benchmark":{"kind":"benchmark","effective_mode":"full","slug":"ds-1000","formal_name":"DS-1000","introduction":"DS-1000 builds 1,000 data-science problems from real StackOverflow questions across seven libraries including NumPy, Pandas and Matplotlib. The problems are perturbed so that recalling the original answer does not solve them.","introduction_ja":"","introduction_en":"","category":"Category not supplied","task_count":null,"acquisition_status":"Acquisition status not supplied","official_url":"https://ds1000-code-gen.github.io/","indexing_mode":"noindex","profile":{"resources":[],"task_format":"","scoring":"","metric":"","size":"","answer_access":"","license":"","citation":"","maintainer":"","released":"","why_hard":"","related":[]}},"task_id":"b70033e5-eb5d-53c2-be25-c561e93452f5","task_key":"default--test--191","task_revision_id":"2","upstream_id":"191","short_description":"I have a dataframe:","config":"default","split":"test","body":"{\"prompt\":\"Problem:\\nI have a dataframe:\\n\\n\\ndf = pd.DataFrame({\\n'A' : ['one', 'one', 'two', 'three'] * 6,\\n'B' : ['A', 'B', 'C'] * 8,\\n'C' : ['foo', 'foo', 'foo', 'bar', 'bar', 'bar'] * 4,\\n'D' : np.random.arange(24),\\n'E' : np.random.arange(24)\\n})\\nNow this will get a pivot table with sum:\\n\\n\\npd.pivot_table(df, values=['D','E'], rows=['B'], aggfunc=np.sum)\\nAnd this for mean:\\n\\n\\npd.pivot_table(df, values=['D','E'], rows=['B'], aggfunc=np.mean)\\nHow can I get sum for D and mean for E?\\n\\n\\n\\n\\nA:\\n<code>\\nimport pandas as pd\\nimport numpy as np\\n\\n\\nnp.random.seed(1)\\ndf = pd.DataFrame({\\n          'A' : ['one', 'one', 'two', 'three'] * 6,\\n          'B' : ['A', 'B', 'C'] * 8,\\n          'C' : ['foo', 'foo', 'foo', 'bar', 'bar', 'bar'] * 4,\\n          'D' : np.random.randn(24),\\n          'E' : np.random.randn(24)\\n})\\n</code>\\nresult = ... # put solution in this variable\\nBEGIN SOLUTION\\n<code>\\n\"}","display_format":"code","language":"","answer_status":"published","assets":[],"source_url":"https://ds1000-code-gen.github.io/","history":"initial import","indexing_mode":"noindex","subproblems":[],"grids":[]}