Manipulating/splicing 列表列表

Manipulating/splicing List of Lists

晚上好, 我可以得到以下代码的建议吗?这是我拥有的:

z_list = df.iloc[0:5000,2]   
n = 1000 #chunk row size
list_df = [z_list[i:i+n].tolist() for i in range(0,z_list.shape[0],n)]

这部分 Python 代码生成一个列表列表,5 个列表,每个列表 1000 行。这将在稍后用于另一个脚本。但是,我想使用 "list_df" 并将每个列表的大小从 1000 减少到它的一小部分(而不是 1000,我可能想要每个列表的前 300 行,所以 5 个 300 行的列表).我尝试过的大多数解决方案似乎不适用于此列表列表设置,因此我不确定该怎么做。任何帮助将不胜感激,谢谢。

使用列表切片?

shortened_sublists = [sublist[:300] for sublist in list_df]