从Python数组中选择元素

时间:2015-07-20 00:08:25

标签: python csv

我使用下面的代码从CSV文件中获取数据并将其放入列表中。输出是列表中的一系列列表,数组。

我想抓取CSV文件中每一行的第一项,并将其分配给变量。我开发的代码有效,但很长。如果可能的话,我有兴趣修改代码的第二部分而不是循环。

with open('player1.csv', 'rb') as f: 

    reader = csv.reader(f) 
    your_list = list(reader)     

print "First Player"
Player1 = your_list[0:1]
Player1name = (Player1[0:1])
Player1name = (Player1name[0])
Player1name = (Player1name[0])
print Player1name

2 个答案:

答案 0 :(得分:1)

如果将每行的每个项目的第一项放入数组而不是为每个项目设置单独的变量呢?

使用list comprehension

这样的事情
player_names = [row[0] for row in your_list]

如果你想打印它们:

for name in player_names:
    print name

答案 1 :(得分:0)

根据提供的描述,假设您要评估存储在string 中的指令,即取自:SDK docs

>>> code = compile('a = 1 + 2', '<string>', 'exec')
>>> exec code
>>> print a
3

但是,每次查找每个CSV文件时,这都是一个繁重的过程。为什么不使用Dictionary并使用键playerNname添加每个变量。例如,playerDict = {'player1Name':var1, 'player2Name':var2, ..., 'playerNName':varN}