我们如何将 numpy 数组拆分为数字的倍数？（例如：数组 1：0-33，数组 2：34-66，数组 3：67-100）答案

【问题标题】：How do we split a numpy array at multiples of a number? (eg: array 1 : 0-33, array 2: 34-66, array 3: 67-100)我们如何将 numpy 数组拆分为数字的倍数？（例如：数组 1：0-33，数组 2：34-66，数组 3：67-100）
【发布时间】：2020-03-08 19:05:35
【问题描述】：

我一直在尝试拆分一个 n 维数组。我想将其拆分为 100//3 (=33) 的倍数

我想将长度为 100 的数组分成 3 组，这样第 1 组来自索引 0-33 第 2 组来自 34-66 第 3 组来自 67-100

这是我尝试过的。

numberblocks=3
array=np.arange(17)
splitat=len(array)//3
for i in range(1,numberblocks):
    np.split(array, splitat*i)

但是，这不会产生任何输出。这个逻辑有什么问题吗？我该如何做到这一点？

【问题讨论】：

我认为这个答案可能会对您有所帮助：stackoverflow.com/questions/312443/…

标签： python numpy numpy-ndarray numpy-slicing

【解决方案1】：

你可以使用：

np.array_split(array, n)

如果可能，将数组 array 拆分为 n 相等的部分，否则最后一个拆分将小于其余部分。

在你的情况下，

>>> array = np.arange(100)
>>> split_arrs = np.array_split(array, 3)
>>> split_arrs
[array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15, 16,
   17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33]), array([34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50,
   51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66]), array([67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83,
   84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99])]
>>> split_arrs[0]
array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15, 16,
   17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33])
>>> split_arrs[1]
array([34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50,
   51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66])
>>> split_arrs[2]
array([67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83,
   84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99])

【讨论】：

如何保存这些以进行进一步操作？
@NewbietoPython 您可以简单地将它分配给一个新对象，然后它将保存一个数组列表。（例如，newarr = np.array_split(array, 3)）
但我的意思是我必须将其拆分为 3 的倍数。我想将长度为 100 的数组拆分为 3 组，以便第 1 组来自索引 0-33 组2 来自 34-66 组 3 来自 67-100 @GoodDeeds
@NewbietoPython 所以对于 arr = [1,2,3,4,5,6]，你想把它拆分成 a1 = [1,4], a2 = [2,5],和 a3 = [3,6]？您的问题并不清楚，正如您所说“我想将长度为 100 的数组拆分为 3 组，以便第 1 组来自索引 0-33 第 2 组来自 34-66 第 3 组来自 67-100”。所以我假设你想要 a1 = [1,2]、a2 = [3,4] 和 a3 = [5,6]。请澄清。
@NewbietoPython 我不明白，这个答案不正是你想要的吗？请查看更新的示例。