Skip to contents
import pandas as pd
import numpy as np

Day 1: The Tyranny of the Rocket Equation

Part 1

grid = np.random.rand(5,2)
df = pd.DataFrame(grid)

def multiply(x):
    return x[0]*x[1]

df['multiply'] = df.apply(multiply, axis = 1)
print(df)
##           0         1  multiply
## 0  0.421962  0.991947  0.418564
## 1  0.404618  0.493774  0.199790
## 2  0.980776  0.037986  0.037255
## 3  0.434080  0.311783  0.135338
## 4  0.521633  0.298229  0.155566