python - numpy 中 [] 和 [[]] 的区别

Question

任何人都可以告诉我们 a[1] 和 a[[1]] 之间的区别(如果 a 是一些 numpy 对象，假设:array)。如果我理解正确，我可以使用 [[1]] 编辑 a 的 1 个元素，但不能使用 [1]...还有其他区别吗？它是否也与 [] 类似，我只是查看一些元素，而 [[]] 我进行复制？如果它有任何重要性，我正在使用 python 3。

Answer 1

当您说 a[1] 时，您将从 a 获取索引为 1 的元素(第二个元素)。另一方面，a[[1]] 并不是一种特殊的语法，它只是表示“获取 a 中由列表 中的索引指示的元素>[1]”。您也可以说 a[[1, 2]]，或使用另一个数组作为索引。实际区别在于 a[1] 将是单个标量元素，而 a[[1]] 将是大小为 1 的数组(假设 a 有一个维度)。

关于赋值，您会发现 a[1] = 2 和 a[[1]] = 2 工作得一样好。这是因为 NumPy 的 broadcasting semantics ，它允许您无缝匹配不同尺寸的对象。但是，a[1] 和 a[[1]] 的实际维度不同，这取决于上下文。

您可以在 NumPy 中阅读更多关于索引的信息 here .