Существует ли лучшее, pythonic способ сделать это?

Поздравляем, у вас очень хороший код. Есть несколько небольших приемов, которые вы можете использовать, чтобы сделать его короче / проще.

Существует изящный тип объекта, называемый defaultdict, который предоставляется модулем коллекций. Вместо того, чтобы проверять, есть ли в adDict ключ adId, вы можете настроить defaultdict, который действует как обычный dict, за исключением того, что он автоматически предоставляет вам пустой set (), когда нет ключа. Таким образом, вы можете изменить

if ( adId in adDict ):
    adDict[adId].add(userId)
else:
    adDict[adId] = set(userId)

на просто

adDict[adId].add(userId)

. Кроме того, вместо

for row in reader:
    adId = row[0]
    userId = row[1]

вы можете сократить это до

for adId,userId in reader:

Редактировать: Как Паркер любезно отмечает в комментариях,

for key, value in adDict.iteritems():

является наиболее эффективным способом перебирать dict, если вы собираетесь использовать оба ключ и значение в цикле. В Python3 вы можете использовать

for key, value in adDict.items():

, поскольку items () возвращает итератор.

#!/usr/bin/env python
import csv
from collections import defaultdict

adDict = defaultdict(set)
reader = csv.reader(open("some.csv"), delimiter=' ')
for adId,userId in reader:
    adDict[adId].add(userId)
for key,value in adDict.iteritems():
    print (key, ',' , len(value))

Вы можете сократить цикл for до следующего:

for row in reader:
  adDict.setdefault(row[0], set()).add(row[1])

Вместо:

for row in reader:
    adId = row[0]
    userId = row[1]

Использовать автоматическую распаковку последовательности:

for (adId, userId) in reader:

В:

if ( adId in adDict ):

Вам не нужны скобки.

Вместо:

if ( adId in adDict ):
    adDict[adId].add(userId)
else:
    adDict[adId] = set(userId)

Используйте defaultdict :

from collections import defaultdict
adDict = defaultDict(set)

# ...

adDict[adId].add(userId)

Или, если преподаватель не разрешил вам использовать другие модули, используйте setdefault () :

adDict.setdefault(adId, set()).add(userId)

При печати:

for key, value in adDict.items():
    print (key, ',' , len(value))

Использование строкового форматирования может быть проще для форматирования:

print "%s,%s" % (key, len(value))

Или, если вы используете Python 3:

print ("{0},{1}".format (key, len(value)))

Поскольку у вас есть только файл с разделителями-пробелами, я бы сделал:

from __future__ import with_statement
from collections import defaultdict

ads = defaultdict(set)
with open("some.csv") as f:
    for ad, user in (line.split(" ") for line in f):
        ads[ad].add(user)

for ad in ads:
    print "%s, %s" % (ad, len(ads[ad]))

Единственные изменения I 'd make извлекают сразу несколько элементов из считывателя и используют строковое форматирование для операторов печати. ​​

import csv

adDict = {}
reader = csv.reader(open("some.csv"), delimiter=' ')
# Can extract multiple elements from a list in the iteration statement:
for adId, userId in reader: 
    if ( adId in adDict ):
        adDict[adId].add(userId)
    else:
        adDict[adId] = set(userId)

for key, value in adDict.items():
    # I believe this gives you more control over how things are formatted:
    print ("%s, %d" % (key, len(value)))

Всего несколько кусочков:

Для извлечения списка строк в переменные:

adId, userId = row

Оператор if не требует фигурных скобок:

if adId in adDict:

Вы могли бы используйте исключения для обработки отсутствующего ключа в слове, но оба способа работают хорошо, например:

try:
    adDict[adId].add(userId)
except KeyError:
    adDict[adId] = set(userId)

Здесь есть несколько отличных ответов.

Один трюк, который мне особенно нравится, - это упростить повторное использование моего кода в будущем вот так

import csv

def parse_my_file(file_name):
     # some existing code goes here
     return aDict

if __name__ == "__main__":
     #this gets executed if this .py file is run directly, rather than imported
     aDict = parse_my_file("some.csv")
     for key, value in adDict.items():
         print (key, ',' , len(value))

Теперь вы можете импортировать парсер csv из другого модуля и получить программный доступ к aDict.