使用功能将列表列表拆分为较小的列表

时间:2018-09-22 13:06:49

标签: list function user-input nested-lists

我有这么长的列表列表(274个元素),这是列表的一小部分:

    ['HR03', 'Jadranska Hrvatska', '8.5'],
    ['HR04', 'Kontinentalna Hrvatska', '8.8'],
    ['HU31', 'Eszak-Magyarorszag', '-2.9'],
    ['HU32', 'Eszak-Alfold', '-0.3'],
    ['HU33', 'Del-Alfold', '0.3'],
    ['IE01', 'Border. Midland and Western', '4.9'],
    ['IE02', 'Southern and Eastern', '4.8'],

我有一个任务,我应该在其中创建一个函数,其中用户输入会创建一个新列表。例如。用户输入“ HU”将创建列表:

 ['HU31', 'Eszak-Magyarorszag', '-2.9'],
 ['HU32', 'Eszak-Alfold', '-0.3']
 ['HU33', 'Del-Alfold', '0.3'],

用户输入“ IE”将创建一个包含所有以“ IE01”,“ IE02”等开头的元素的列表。 我在这里完全迷路了。我不知道该怎么做,我的教科书也没有解释。任何帮助,将不胜感激。

1 个答案:

答案 0 :(得分:0)

具有条件的列表理解应该可以解决问题

lst = # data...
prefix = input("Enter a prefix: ")
result = [x if x[0].startswith(prefix) for x in lst]