擅长:python、mysql、java
<p>是的:</p>
<pre><code>import numpy as np
import pandas as pd
data = pd.DataFrame({'x': np.random.randint(1,3,10),
'y': np.random.randint(1,3,10)})
print(data)
</code></pre>
<p>输出:</p>
^{pr2}$
<pre><code>def isThree(x, y):
return (x + y == 3)
print(data[isThree(data['x'], data['y'])].head())
</code></pre>
<p>输出:</p>
<pre><code> x y
0 1 2
1 2 1
2 2 1
3 1 2
4 2 1
</code></pre>