python - 如何将一个 numpy 数组拆分为多个批次，其中每个批次中的行数是用户指定的

Question

例如，我有一个 110 行的 numpy 数组。

如何拆分此数组，以便每个拆分/批次有 20 行，其中 20 由用户指定？这还需要考虑到最后一个只有 10 行的拆分/批处理。

我知道有类似的 numpy 函数，np.array_split()但此函数不允许您指定每个拆分/批次应该有多少行（它只允许您指定您想要的拆分/批次数量）。

提前谢谢了。

标签： pythonnumpybatch-processing