# -- Code Cell --
import pandas as pd
df = pd.read_csv('train.csv')

# -- Code Cell --
wins = df[df['reward'] == 1000.0]
wins['next_state'].value_counts()

# -- Code Cell --
DARWIN = 777
darwin_row = DARWIN // 32  
darwin_col = DARWIN % 32   

# -- Code Cell --
def best_action(s):
    dr = darwin_row - s // 32  
    dc = darwin_col - s % 32    
    if abs(dr) >= abs(dc):
        return 1 if dr > 0 else 0 
    else:
        return 3 if dc > 0 else 2   

# -- Code Cell --
test = pd.read_csv("test.csv")

# -- Code Cell --
submission = pd.DataFrame({
    "state": test["state"],
    "action": test["state"].apply(best_action)
})
submission.to_csv("submission.csv", index=False)

# -- Code Cell --
!python eval.py

# -- Code Cell --
