python - 使用来自文件的输入创建字典

Question

我有一个格式如下的文件:

Barcelona   2015,2016,2017
Real Madrid   2010
Napoli   2007,2009
Bayern Munich   2008,2009,2010,2011,2012,2013

我想将其保存到包含团队的字典中，然后是包含数字的列表。我该怎么做？我在拆分时遇到了一些困难，因为有些团队的名字更大。

Answer 1

这是 Pandas 解决方案，假设使用 4 个空格分隔符。

import pandas as pd
from io import StringIO

mystr = StringIO("""Barcelona   2015,2016,2017
Real Madrid   2010
Napoli   2007,2009
Bayern Munich   2008,2009,2010,2011,2012,2013""")

df = pd.read_csv(mystr, delimiter='   ', header=None, names=['Club', 'Years'])

df['Years'] = [list(map(int, x)) for x in df['Years'].str.split(',')]
d = df.set_index('Club')['Years'].to_dict()

结果

{'Barcelona': [2015, 2016, 2017],
 'Bayern Munich': [2008, 2009, 2010, 2011, 2012, 2013],
 'Napoli': [2007, 2009],
 'Real Madrid': [2010]}

解释

• 使用适当的分隔符和名称列读取文件。
• 用逗号分隔并通过列表理解将每个元素映射到整数类型。
• 创建由 Club 索引的系列并使用 .to_dict() 输出字典。