擅长:python、mysql、java
<p>你可以做<code>transform</code>:</p>
<pre><code>s = (data.set_index('roi') # filter `roi` out of later comparison
.lt(50).any(1) # check > 50 on all columns
.groupby('roi') # groupby
.transform('mean') # compute the mean
.lt(0.5) # make sure mean > 0.5
.values
)
data[s]
</code></pre>
<p>输出:</p>
<pre><code> roi s b1 b2
3 5 60 40 60
4 5 60 60 60
5 5 60 60 60
</code></pre>