text_plain_output_1.png
cols_with_missing = [col for col in X_train.columns if X_train[col].isnull().any()]
cols_with_missing

test_cols_with_missing = []
for col in X_train.columns:
    if X_train[col].isnull().any():
        test_cols_with_missing.append(col)
test_cols_with_missing