Update code to remove NaN values

Author: Jared Weed

projects/titanic_survival_exploration/titanic_survival_exploration.zip

@@ -64,14 +64,19 @@ def survival_stats(data, outcomes, key, filters = []):
     # Apply filters to data
     for condition in filters:
         all_data = filter_data(all_data, condition)
+
+    # Create outcomes DataFrame
     all_data = all_data[[key, 'Survived']]
 
     # Create plotting figure
     plt.figure(figsize=(8,6))
 
     # 'Numerical' features
     if(key == 'Age' or key == 'Fare'):
-       
+        
+        # Remove NaN values from Age data
+        all_data = all_data[~np.isnan(all_data[key])]
+        
         # Divide the range of data into bins and count survival rates
         min_value = all_data[key].min()
         max_value = all_data[key].max()