擅长:python、mysql、java
<p>当你可以使用向量化操作时,Pandas将获得最大的效率胜利,但我认为这个问题需要迭代。使用熊猫的解决方案:</p>
<pre><code>import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randint(0,2,100)*2-1, columns=['TossResults'])
initial_stake = 1
df['Stake'] = initial_stake
for i in xrange(1,df.shape[0]):
if df.TossResults[i-1] == -1:
df.Stake[i] = 2 * df.Stake[i-1]
</code></pre>