输入指向元组列表中某些元素的点

时间:2013-03-19 23:46:13

标签: input python-3.x tuples

我有一个元组列表如下:

EXCHANGE_DATA = [('AIA', 'Auckair', 1.50),
                 ('AIR', 'Airnz', 5.60),
                 ('AMP', 'Amp',3.22), 
                 ('ANZ', 'Anzbankgrp', 26.25),
                 ('ARG', 'Argosy', 12.22),
                 ('CEN', 'Contact', 11.22),
                 ('CNU', 'Chorus',3.01),
                 ('DIL', 'Diligent', 5.3),
                 ('DNZ', 'Dnz Property', 2.33),
                 ('EBO', 'Ebos', 1.1)]

我想要做的是让用户输入这些元组中的一系列代码(每个元组中的第一个元素),用逗号分隔(例如AIA,AIR,AMP)并让程序返回整个元组输入是指。例如,如果输入是AIA,AIR,AMP,程序将返回:

('AIA', 'Auckair', 1.50)
('AIR', 'Airnz', 5.60)
('AMP', 'Amp',3.22)

我对编程比较陌生,所以任何帮助都会受到赞赏。

1 个答案:

答案 0 :(得分:0)

您应该创建一个帮助字典。这是避免嵌套循环的好方法。

>>> EXCHANGE_DATA = [('AIA', 'Auckair', 1.50),
...                  ('AIR', 'Airnz', 5.60),
...                  ('AMP', 'Amp',3.22), 
...                  ('ANZ', 'Anzbankgrp', 26.25),
...                  ('ARG', 'Argosy', 12.22),
...                  ('CEN', 'Contact', 11.22),
...                  ('CNU', 'Chorus',3.01),
...                  ('DIL', 'Diligent', 5.3),
...                  ('DNZ', 'Dnz Property', 2.33),
...                  ('EBO', 'Ebos', 1.1)]
>>> inp = "AIA,AIR,AMP"     # could come from input()
>>> exchange_dict = {i[0]:i for i in EXCHANGE_DATA}
>>> [exchange_dict[x] for x in inp.split(',')]
[('AIA', 'Auckair', 1.5), ('AIR', 'Airnz', 5.6), ('AMP', 'Amp', 3.22)]

如果输入可以包含额外的空格,您可以像这样删除它们

>>> [exchange_dict[x.strip()] for x in inp.split(',')]
[('AIA', 'Auckair', 1.5), ('AIR', 'Airnz', 5.6), ('AMP', 'Amp', 3.22)]