np.random.choice应该强制吗？

Question

我只是注意到，如果在包含字符串和整数的列表上使用np.random.choice，那么当返回整数时，它们将被强制转换为字符串。 这是预期的行为吗？

例如

>>>numpy.random.choice([1,2,3,4]) 1

但

>>>numpy.random.choice(['a',1,2,3,4]) '1'

我想我可以做my_list[np.random.choice(range(len(my_list)))] ，但这看起来很丑。

Answer 1

这不是随机选择所独有的。 NumPy方法可用于NumPy数组，如果给出了Python列表，则它们将在执行其他操作之前将其转换为数组。 转换结果为：

>>> np.array(['a', 1, 2, 3, 4])
array(['a', '1', '2', '3', '4'],
      dtype='<U1')

一字符的Unicode字符串数组。 从此数组中进行选择。