2

希望我在标题中已经解释得足够好，但这是我所拥有的：

输入数据：

Mushroom Kingdom, Mario

Hyrule, Link

Mushroom Kingdom, Bowser

Zebes, Samus

Zebes, Metroid

我想运行这样的东西，

# The next three lines establish that I'll be reading proc as a file
import subprocess
cmd = 'external command that returns the above data'
proc = subprocess.Popen(cmd, stdout=subprocess.PIPE)

homeworlds = {}

while True:
  line = proc.stdout.readline().split(',')
  if line:
    # If line isn't empty
    homeword = line[0]
    person = line[1]

    homeworlds[homeword] = list.append[person] # Good logic? Bad syntax?

  else:
   break

目标是我将能够调用：

print homeworlds['Mushroom Kingdom']

并返回列表

Mario, Bowser
4

6 回答 6

3

在这里使用 a 更容易defaultdict。此外，您当前正在调用split()没有任何参数。你真正想做的是在,. 所以你应该打电话split(',')。

所以你的代码应该看起来像这样：

homeworlds = collections.defaultdict(list)
while True:
  line = proc.stdout.readline().split(',')
  if line:
    # If line isn't empty
    homeworld = line[0].strip()
    person = line[1].strip()

    homeworlds[homeworld].append(person)

  else:
    break
于 2012-07-30T16:51:01.193 回答
2 
 if not homeword in homeworlds:
    homeworlds[homeword] = list()
 homeworlds[homeword].append(person)
于 2012-07-30T16:48:31.807 回答
1

利用defaultdict

from collections import defaultdict
homeworlds = defaultdict(list)

homeworlds[homeword].append(person)
于 2012-07-30T16:51:47.570 回答
0

这是一个一致而优雅的解决方案（部分灵感来自其他答案）：

#!/bin/env python
from collections import defaultdict

lines = """Mushroom Kingdom, Mario

Hyrule, Link

Mushroom Kingdom, Bowser

Zebes, Samus

Zebes, Metroid
"""

homeworlds = defaultdict(list)

for line in lines.splitlines():
    if line and ',' in line:
        key, value = line.split(',')
        homeworlds[key].append(value.trim())

print(homeworlds)
于 2012-07-30T16:59:49.883 回答
0 
Python 2.7.3rc2 (default, Apr 22 2012, 22:30:17) 
[GCC 4.6.3] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> input_data = [('Mushroom Kingdom', 'Mario'),
...  ('Hyrule', 'Link'),
...  ('Mushroom Kingdom', 'Bowser'),
...  ('Zebes', 'Samus'),
...  ('Zebes', 'Metroid')]
>>> homeworlds = {}
>>> for homeword,person in input_data:
...   homeworlds.setdefault(homeword, []).append(person)
... 
>>> homeworlds
{'Hyrule': ['Link'], 'Zebes': ['Samus', 'Metroid'], 'Mushroom Kingdom': ['Mario', 'Bowser']}
>>> print(homeworlds['Mushroom Kingdom'])
['Mario', 'Bowser']
>>> print("{0}".format(', '.join(homeworlds['Mushroom Kingdom'])))
Mario, Bowser

在您的情况下，换入 for 循环： 'input_data' for 'line'

于 2012-07-30T17:07:42.843 回答
0

尝试这个：

if homeword not in homeworlds:
    homeworlds[homeword] = []
homeworlds[homeword].append(person)

这将检查是否有homewordin的条目homeworld。如果没有，它会在那里添加一个列表。然后我们确定我们有一个列表，homeworld[homeword]所以我们可以调用.append(person)它。

于 2012-07-30T16:49:54.117 回答