擅长:python、mysql、java
<p>你可以这样做:</p>
<pre><code>In [114]: %paste
def calculate_tax(income, tax1=0.13, tax2=0.2, thresh=1000):
s = pd.Series([income] * 12)
return (s.mul(tax1 + s.cumsum().gt(thresh)
.shift().fillna(False)
.mul(tax2-tax1))
.cumsum()
.iloc[-1])
## End pasted text
In [115]: calculate_tax(150, 0.13, 0.2, 1000)
Out[115]: 286.5
In [161]: calculate_tax(150)
Out[161]: 286.5
</code></pre>