操作/拼接列表列表

晚上好,     请问以下代码的建议?这是我所拥有的:

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

评论
  • psed
    psed 回复

    使用列表切片?

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