如何在Python中截取嵌套列表的前n个子列表

我有一个多层嵌套的列表结构（示例如下），需要实现提取外层列表中的前n个子列表，比如当n=6时，返回外层列表的前6个元素。

示例嵌套列表结构：

[[[['Groningen',
    'GM0014',
    231299,
    0.49691957163671263,
    0.5030804283632874,
    0.1246092719812883,
    0.2176490170731391,
    0.2934988910457892,
    0.22006147886501887,
    0.14418134103476452,
    0.6526227956022291,
    0.2415531411722489,
    0.07018620919243057,
    0.035637854033091367],
   ['Amsterdam',
    'GM0363',
    862965,
    0.4957176710527078,
    0.5042823289472922,
    0.14554703840827843,
    0.1257142526058415,
    0.36046421349649177,
    0.24314775222633594,
    0.12512674326305237,
    0.6432080095948272,
    0.2436031588766636,
    0.08585516214446703,
    0.027333669384042227],
   ['Utrecht',
    'GM0344',
    352866,
    0.48986584142422335,
    0.5101341585757766,
    0.17020908786904945,
    0.1524119637482784,
    0.36236701750806255,
    0.21132667924934678,
    0.10368525162526285,
    0.6469027902943327,
    0.26626538119286075,
    0.05933697210839242,
    0.02749485640441414],
   ['Nijmegen',
    'GM0268',
    176731,
    0.4842500749726986,
    0.5157499250273013,
    0.1371293095155915,
    0.1775127170671812,
    0.2857506606084954,
    0.24216464570448873,
    0.15744266710424318,
    0.6046251082153102,
    0.2767256451895819,
    0.08034809965427683,
    0.038301146940830984],

解决方案 #

在Python中，直接使用列表切片即可实现需求，语法简单高效：

# 假设你的嵌套列表变量名为data
n = 6
# 提取外层列表的前n个子列表
result = data[:n]

说明

• 切片[:n]会从列表起始位置开始，截取到索引为n的位置（不包含该索引），也就是取外层列表的前n个元素
• 如果外层列表的元素总数小于n，切片会自动返回整个外层列表，不会抛出异常

内容的提问来源于stack exchange，提问作者Nikos Koutas