操作/拼接列表列表

Question

晚上好，请问我可以就以下代码获得建议吗？ 这是我所拥有的：

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 行的列表）。 我尝试过的大多数解决方案似乎都不适用于此列表设置，所以我不确定该怎么做。 任何帮助将不胜感激，谢谢。

Answer 1

使用列表切片？

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