day 26 - comprehensions

025/main.py

@@ -18,10 +18,9 @@ found_states = []
 while len(found_states) < 50:
     guess = str(screen.textinput(f"{states_remaining} states to find", "Find a state:"))
     if guess == "None":
-        with open("missing_states.csv", "w") as f:
-            for a_state in data.state:
-                if a_state not in found_states:
-                    f.write(f"{a_state}\n")
+        missing_states = [ s for s in data.states if s not in found_states ]
+        output = pd.DataFrame(missing_states)
+        output.to_csv("missing_states.csv")
         exit()
     
     state = data[data.state == guess.title()]