def write_number_of_positive_traids(data, definition, def_args, out_dir):
    def relationship_def(data, year, A, B, def_args1):
        return definition(data, year, A, B, def_args) == POSITIVE_LINK

    all_years_data = {}
    years = data.all_years
    countries_list = data.countries()

    for year in years:
        single_year_data = {}
        with open(out_dir + '%s.csv' % year, 'wb') as csvfile:
            out_file = csv.writer(csvfile, delimiter=',', quotechar='"', quoting=csv.QUOTE_ALL)
            out_file.writerow(['Country', 'Number of positive traids'])
            relationship_matrix = get_relationship_matrix(data, year, relationship_def, {})
            relationship_matrix_cube = __matrix_cube(relationship_matrix)
            for C in countries_list:
                number_of_positive_traids = number_of_traids_for_a_country(relationship_matrix_cube, C)
                out_file.writerow([C, number_of_positive_traids])
                single_year_data[C] = number_of_positive_traids
        all_years_data[year] = single_year_data

    with open(out_dir + 'all-years.csv', 'wb') as csvfile:
        out_file = csv.writer(csvfile, delimiter=',', quotechar='"', quoting=csv.QUOTE_ALL)
        header = ['*']
        for C in countries_list:
            header.append(C)
        out_file.writerow(header)
        for year in years:
            row = [year]
            for C in countries_list:
                row.append(all_years_data[year][C])
            out_file.writerow(row)
예제 #2
0
def table1(data, year, definition, def_args):
    def link_exists_def(data, year, A, B, def_args1):
        if 'World' in [A, B]: return 0
        return 0 if definition(data, year, A, B, def_args) == NO_LINK else 1

    edges_count, positives, negatives, _ = sign_distributions(data, year, definition, def_args)
    return {'Name': 'Table1',
            'Nodes': len(data.countries()),
            'Edges': edges_count,
            '+ edges': positives * 100.0 / edges_count,
            '- edges': negatives * 100.0 / edges_count,
            'Traids': number_of_traids(__matrix_cube(get_relationship_matrix(data, year, link_exists_def, {})))
    }