Titanic Code

import os
import csv
import numpy as np


def import_data(filename):
    cwd = os.getcwd()
    filepath = cwd+'\\' + filename
    X = []
    y = []
    with open(filepath,'r') as f:
        data = csv.reader(f)
        next(data)
        for sample in data:
            x_temp = []
            x_temp.append(int(sample[0]))
            x_temp.append(int(sample[2]))
            if sample[4] == "male":
                x_temp.append(1)
            else:
                x_temp.append(0)
            if sample[5] == "":
                x_temp.append(float("NaN"))
            else:
                x_temp.append(float(sample[5]))
            x_temp.append(int(sample[6]))
            x_temp.append(int(sample[7]))
            x_temp.append(float(sample[9]))
            if sample[11] == 'C':
                x_temp.append(0)
            elif sample[11] == 'Q':
                x_temp.append(1)
            else:
                x_temp.append(2)
            X.append(x_temp)
            y.append(int(sample[1]))
    return X,y


filename = input("Enter Filename: ")
X,y = import_data(filename)