操作/拼接列表列表

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]