def read_csv()

in phasic_policy_gradient/graph_util.py [0:0]

• 15 lines of code
• 10 McCabe index (conditional complexity)

def read_csv(filename, key_name):
    with open(filename) as csv_file:
        csv_reader = csv.reader(csv_file, delimiter=',')
        key_index = -1

        values = []

        for line_num, row in enumerate(csv_reader):
            row = [x.lower() for x in row]
            if line_num == 0:
                idxs = [i for i, val in enumerate(row) if val == key_name]
                key_index = idxs[0]
            else:
                values.append(row[key_index])

        values = [np.nan if x == '' else float(x) for x in values]
        values = [x for x in values if not np.isnan(x)]

    return np.array(values, dtype=np.float32)