使用 Numpy 数组创建唯一数组

Question

你能创建一个包含所有唯一值的 numpy 数组吗？

myArray = numpy.random.random_integers(0,100,2500)
myArray.shape = (50,50)

所以这里我有一个给定的随机 50x50 numpy 数组，但我可以有非唯一值。有没有办法确保每个值都是唯一的？

谢谢

更新：

我创建了一个基本函数来生成一个列表并填充一个唯一的整数。

        dist_x = math.sqrt(math.pow((extent.XMax - extent.XMin), 2))
        dist_y = math.sqrt(math.pow((extent.YMax - extent.YMin),2))
        col_x = int(dist_x / 100)
        col_y = int(dist_y / 100)
        if col_x % 100 > 0:
            col_x += 1
        if col_y % 100 > 0:
            col_y += 1
        print col_x, col_y, 249*169
        count = 1
        a = []

        for y in xrange(1, col_y + 1):
            row = []
            for x in xrange(1, col_x + 1):
                row.append(count)
                count += 1
            a.append(row)
            del row

        numpyArray = numpy.array(a)

有没有更好的方法来做到这一点？

谢谢

Answer 1

从集合中获取唯一随机样本的最方便的方法可能是np.random.choice 和replace=False。

例如：

import numpy as np

# create a (5, 5) array containing unique integers drawn from [0, 100]
uarray = np.random.choice(np.arange(0, 101), replace=False, size=(5, 5))

# check that each item occurs only once
print((np.bincount(uarray.ravel()) == 1).all())
# True

如果replace=False 您要从中采样的集合当然必须至少与您尝试抽取的样本数量一样大：

np.random.choice(np.arange(0, 101), replace=False, size=(50, 50))
# ValueError: Cannot take a larger sample than population when 'replace=False'

---

如果您要查找的只是 1 和数组中元素数之间的整数的随机排列，您也可以像这样使用 np.random.permutation：

nrow, ncol = 5, 5
uarray = (np.random.permutation(nrow * ncol) + 1).reshape(nrow, ncol)

Answer 2

只需使用 replace = False。

import numpy as np


def generate():
    global x
    powerball = np.random.randint(1,27)
    numbers = np.random.choice(np.arange(1, 70), replace=False, size=(1, 5))
    x = numbers, powerball
    return x

generate()