python中csv库用法_如何使用for循环和Python中的csv库遍历列？

import csv

with open('in.csv')as f:

head = next(f)

# decide delimiter by what is in header

delim = "," if "," == head[0] else " "

# need to filter empty strings

head = filter(None, head.rstrip().split(delim))

# skipinitialspace must be set as you have two spaces delimited

reader = csv.reader(f,delimiter=delim, skipinitialspace=True)

# transpose rows

zipped = zip(*reader)

# skip first column

strings = next(zipped)

# sum each column

print([sum(map(int,col)) for col in zipped])

[0, 16, 22, 22, 0, 20]

要创建将标题与colunm sums相匹配的dict，您可以这样做：print(dict(zip(list(head), (sum(map(int,col)) for col in zipped))))

哪些输出：{'u': 20, 'w': 22, 'x': 22, 'z': 0, 'y': 16, 'v': 0}

我使用python3来完成以上所有操作，如果您使用python2，请替换为：zip -> itertools.izip

filter -> itertools.izip

map -> itertools.imap

Python2代码：import csv

from itertools import izip, imap, ifilter

with open('in.csv')as f:

head = next(f)

# decide delimiter by what is in header

delim = "," if "," == head[0] else " "

# need to filter empty strings

head = ifilter(None, head.rstrip().split(delim))

# skipinitialspace must be set as you have two spaces delimited

reader = csv.reader(f,delimiter=delim, skipinitialspace=True)

# transpose rows

zipped = izip(*reader)

# skip first column

strings = next(zipped)

# sum each column

print([sum(imap(int,col)) for col in zipped])

输出：[0, 16, 22, 22, 0, 20]

如果你正在做很多这样的工作，那么熊猫尤其是pandas.read_csv可能是有用的，下面是一个非常基本的例子，一些熊猫大师可能会希望添加到其中：import pandas as pd

df = pd.read_csv("in.csv")

print(df.sum())

Unnamed: 0 abcd

z 0

y 16

x 22

w 22

v 0

u 20

dtype: object