擅长:python、mysql、java
<p>您可以创建一个定义组的字典。你知道吗</p>
<pre><code>mapping = {**dict.fromkeys([1, 2, 3], 'P'), **dict.fromkeys([4, 5], 'E')}
</code></pre>
<p>然后,如果您传递字典或可调用到<code>groupby</code>,它会自动映射到索引上。那么,让我们将索引设置为<code>Class</code></p>
<pre><code>d = df.set_index('Class').groupby(mapping).agg(['sum', 'mean']).sort_index(1, 1)
</code></pre>
<p>最后,我们做了一些调整,以获得您指定的列名。你知道吗</p>
<pre><code>rename_dict = {'sum': 'Total', 'mean': 'Average'}
d.columns = d.columns.map(lambda c: f"{rename_dict[c[1]]}{c[0].title()}")
d.rename_axis('Class').reset_index()
Class TotalPrice TotalDemand AveragePrice AverageDemand
0 E 87 45 43.5 22.5
1 P 186 38 46.5 9.5
</code></pre>