Как получить суммы в CSV-файле python

Question

[введите описание изображения здесь] [1] [введите описание изображения здесь] [2] У меня есть файл CSV с 14 категориями с 303 строками Мне нужно найти сумму для каждой категории. Что я до сих пор, к сожалению, не очень помогает в том, как найти сумму каждой категории по отдельности и сокращенно 14 категорий в порядке возраста, gen, cpt, rbp, sc, fbs, rer, mhra, eia, с.о., SPSD, NMV, дт, HD

import string 
with open("train.csv", "r") as f: 
    for hd in f.read():13 

первые несколько строк, как это

ABCDE ф GHIJKLMN

63 1 1 145 233 1 2 150 0 2,3 3 0 6 0

67 1 4 160 286 0 2 108 1 1,5 2 3 3 2

67 1 4 120 229 0 2 129 1 2,6 2 2 7 1

Answer 1

Вы можете карту каждый ряд поплавка, транспозиции и сумму каждый столбец делая Dict с (column_name, column_sum) спариваний:

import csv 

with open("train.csv", "r") as f: 
    r = csv.reader(f) 
    # pull first row i.e A,B,C,D... to use ad the keys 
    keys = next(r) 
    # iterate ober the rows mapping to float and transpose 
    # with zip(*... so rows become columns, one per key 
    summed_dict = dict(zip(keys, map(sum, zip(*(map(float, row) for row in r))))) 

Для вашего ввода образца:

A,B,C,D,E,f,G,H,I,J,K,L,M,N 
63,1,1,145,233,1,2,150,0,2.3,3,0,6,0 
67,1,4,160,286,0,2,108,1,1.5,2,3,3,2 
67,1,4,120,229,0,2,129,1,2.6,2,2,7,1 

Выходы:

{'K': 7.0, 'D': 425.0, 'B': 3.0, 'G': 6.0, 'f': 1.0, 'N': 3.0, 'M': 16.0, 'C': 9.0, 'A': 197.0, 'E': 748.0, 'L': 5.0, 'H': 387.0, 'I': 2.0, 'J': 6.4} 

Вы можете использовать DictReader, но здесь нет никаких реальных преимуществ.

Answer 2

Python имеет действительно полезный модуль «csv».

import csv 

with open("train.csv", "r") as f: 
    reader = csv.reader(f) 
    for row in reader: 
     # work on each row as a list 

Существует также удобный DictReader, который анализирует первую строку CSV в качестве заголовков строк, а затем дает каждую строку в качестве словаря с установленными ключами к коллекторам.

Теперь, когда данные находятся в удобном для употребления формате, суммирование должно быть простым.

Answer 3

Numpy может быть вашим другом:

import numpy as np 

# load array from csv 
arr = np.loadtxt('data.csv', dtype=float, delimiter=';', skiprows=0) 

n = -1 
# get n-th row by fany indexing 
row = arr[n,:] 
print row 
print np.sum(row) 

m = 2 
# get m-th col by fancy indexing 
col = arr[:,m] 
print col 
print np.sum(col) 

это в значительной степени само за себя. для получения дополнительной информации: read the docs