按索引从列表中选择，但将其他关联项目保留在列表中

Question

如果我有2dlist这样的点...

myList=[['ab','0_3','-1','1'],['bm','2_1','-3','2'],['am','4_1','-1','3'],...]]

例如， '-1','1'是x，y坐标（最后2列），我想只显示列表的一部分，其索引是这样的...

[0] ab    # I don't want to display the other list items, just the index and first col
[1] bm
[2] am

...因此用户可以通过索引号选择其中一个作为起点。 我想我可以这样列举...

for row in enumerate(myList):
     i, a= row
     col= (i, a[0])
     newList.append(col)
     print cols

但是，一旦我要求用户选择一个即。 用户选择'0'并设置变量origin='ab' ，我如何获取与原点关联的x，y（或[2]，[3]）列用作原点坐标（我需要与列表的其余部分进行比较）？

使用这种方法，我可以以某种方式使用分配给所选点的变量。 origin = ab ，然后得到它的x，y并将它们分配给x1，y1 ...因为enumerate给出了两个元组（i，a）并将它们附加到newList ，这是我在newList拥有的newList ，或者我也可以追加其他列但不显示它们？ 我希望我的解释很清楚...我只有糟糕的代码自动取款机

---

因此，我终于将大部分工作与...

import csv
myList=[]    
try:
    csvr = open('testfile.csv','r')
    next(csvr, None)
    theList = csv.reader(csvr)

    for row in theList:
        myList.append(row)

    for row in enumerate(myList):
        i, a = row
    print i, a[0]
except IOError:
    print 'Error!!!'


try:
   choice = raw_input("Select a set: ") # Can also enter 'e'
   if choice=='e'
      print 'exiting'
   else: 
        pass 
    user_choice = myList[int(choice)]   
    name, id, x, y= user_choice     
    print name, id, 
    return float(x), float(y)    
except:     
   print 'error'

它会按预期打印，现在我可以返回x，y，这很好，但是它一直在提示我输入数字。 有什么建议么？

Answer 1

1）保持列表格式，您可以使用用户的选择作为列表索引：

myList = [['ab','0_3','-1','1'],['bm','2_1','-3','2'],['am','4_1','-1','3']]

for row in enumerate(myList):
     i, a = row
     print i, a[0]


choice = int(raw_input("Select a set: "))
user_choice = myList[choice]

name, id, x, y = user_choice

print name, id, float(x) + float(y)  # use float() to convert strings to floats

样本输出：

~ $ python tester.py
0 ab
1 bm
2 am
Select a set: 1
bm 2_1 -1.0