Python:对应的CSV行值

时间:2017-03-06 16:31:29

标签: python csv for-loop corresponding-records

对于代码中指定的两列(IP和平台)的每个对应值,我想连接它们,然后计算这个连接字符串在整个csv数据库中出现的次数。

此代码目前需要大约一个小时才能在命令行中查看1000行数据,这似乎是double-for循环的结果。如果不出意外,我怎样才能将每个相应IP的平台附加到users_country_platform列表而不使用这个双for循环?

import sys
import csv
from collections import Counter

users_country_platform = []
with open(sys.argv[1], 'r') as UserLog:
    IPs = [row['client_ip_address'] for row in csv.DictReader(UserLog)]
    UserLog.seek(0)
    platforms = [row['platform'] for row in csv.DictReader(UserLog)]
    for platform in platforms:
        for IP in IPs:
            if IP != 'NULL' and platform != 'NULL' or 'None':
               users_country_platform.append(str(response.country.iso_code) + ', ' + platform)

另一种解决方案是在国家/地区密钥中建立子阵列,以便输出格式为[country:platform:count],但我认为我需要一种不同的计数方法。对这两种方法的建议将不胜感激。

1 个答案:

答案 0 :(得分:1)

enter image description here

我使用了一个非常简单的示例,但希望您可以使用以下内容并将其用于您的代码。采用上面的csv格式,请考虑以下内容:

import csv
import os
cwd = os.getcwd()

counter = {}
with open(cwd+'\\test.csv', 'rb') as file_in:
    reader = csv.DictReader(file_in, restval=None, restkey=None, dialect='excel')
    for row in reader:
        conc = row['ip'] + row['platform']
        counter.setdefault(conc, 0)
        counter[conc] += 1

print counter

希望这显示如何只用一个循环计算两列的唯一出现次数!