numpy 在Python中使用操作索引数组

2j4z5cfb 于 2023-05-07 发布在 Python

关注(0)|答案(1)|浏览(123)

我正在寻找一种方法来创建一个索引数组，其中包括对一些条目的操作。
例如，我有一组点，我想提取第一个，第二个和第一个减去第二个点。

import numpy as np

p = np.array([1.1, 2.2, 3.3])

magic_idx_array = [0, 1, [0]-[1]]

print(p[magic_idx_array])

这给了我一个错误，但我希望：

array([1.1, 2.2, -1.1])

我知道我可以在这里使用一个函数，但是，我想保持它的简单和快速。

1条答案

你所做的在语法上是错误的：
你应该做：

import numpy as np

p = np.array([1.1, 2.2, 3.3])

[p[0], p[1], p[0]-p[1]]

#output
[1.1, 2.2, -1.1]

或

np.array([p[0], p[1], p[0]-p[1]])

#output
array([ 1.1,  2.2, -1.1])