在numpy中随机播放数组的一部分

Question

我有一个 numpy 数组，我想改组它的一部分。例如，使用以下数组：

import numpy as np
import random

a = np.arange(15)
# => array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14])

我想做：

shuffle_parts(a, [(0, 3), (10, 13)])
# => array([ 2,  0,  1,  3,  4,  5,  6,  7,  8,  9, 12, 11, 10, 13, 14])
#            ^^^^^^^^^                              ^^^^^^^^^^
#            Shuffle those 3 values                 and those 3 values

以下内容将对所有数组进行洗牌：（不是我想要的）

random.shuffle(a) 
# => array([10, 11,  8,  1, 13,  5,  9, 14,  4,  7,  2, 12,  3,  0,  6])

一种方法是像这样使用拆分/连接：

splits = np.split(a, 5)
random.shuffle(splits[0])
random.shuffle(splits[3])
np.concatenate(splits)
# => array([ 2,  0,  1,  3,  4,  5,  6,  7,  8, 11, 10, 9, 12, 13, 14])
#            ^^^^^^^^^                          ^^^^^^^^^^
#            Correctly shuffled                 Shuffled but off by 1 index

这几乎就是我想要的。我的问题：

• 我可以写shuffle_parts，其中索引是自定义的（具有任意索引的部分，不限于模数，以及长度可变的部分）
• numpy 中是否有我错过的方法可以帮助我做到这一点？

Answer 1

可以直接做：

>>> import numpy as np
>>> import random
>>> a = np.arange(15)
>>> s=3
>>> f=7
>>> random.shuffle(a[s:f])
>>> a
array([ 0,  1,  2,  5,  4,  3,  6,  7,  8,  9, 10, 11, 12, 13, 14])

索引直接引用数据，使这成为可能。

Answer 2

numpy slices are views 在下面的数据上；所以你可以直接洗牌：

import numpy as np
import random

a = np.arange(15)

random.shuffle(a[0:3])
random.shuffle(a[10:13])
print(a)
# [ 2  0  1  3  4  5  6  7  8  9 12 10 11 13 14]

您可以使用slice 实现您的shuffle_parts 函数，然后：

def shuffle_parts(array, slices):
    for s in slices:
        random.shuffle(a[slice(*s)])

shuffle_parts(array=a, slices=((0, 3), (10, 13)))

或（取决于您希望如何将切片传递给您的函数）：

def shuffle_parts(array, slices):
    for s in slices:
        random.shuffle(a[s])

shuffle_parts(array=a, slices=(slice(0, 3), slice(10, 13)))

我个人更喜欢第二个版本（这样你也可以洗牌偶数索引：shuffle_parts(array=a, slices=(slice(None, None, 2), ))）...