python - 将集合中的元素插入数组(numpy)

Question

我有一组 (S) 数字，我想将这些数字放入数组 (arr) 中。我尝试了这段代码

Arr = np.array(S)

但是我无法访问数组元素，例如如果我尝试

Arr[0]

，我收到此错误:

IndexError: too many indices for array

任何人都可以解释这种方法有什么问题吗？我可以使用其他方法将 set 的元素放入数组并访问它们吗？

谢谢

Answer 1

您首先需要将一组数字转换为列表。

S = {1, 2, 3}

>>> np.array(S)
array(set([1, 2, 3]), dtype=object)

>>> np.array(list(S))
array([1, 2, 3])