用来自lis的随机元素创建numpy数组

Question

是否有一种有效的方法来创建任意的长numpy数组，其中每个维度都由从长度列表中提取的元素组成？列表中的每个元素只能为每个维度绘制一次。

例如，如果我有列表l = ['cat', 'mescaline', 'popcorn']，我希望能够创建一个数组array([['cat', 'popcorn'], ['cat', 'popcorn'], ['mescaline', 'cat']])，例如通过键入np.random.pick_random(l, (3, 2), replace=false)。

谢谢你。

Answer 1

>>> import numpy
>>> l = numpy.array(['cat', 'mescaline', 'popcorn'])
>>> l[numpy.random.randint(0, len(l), (3, 2))]
array([['popcorn', 'mescaline'],
       ['mescaline', 'popcorn'],
       ['cat', 'cat']], 
      dtype='|S9')