操作/拼接列表列表

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]