我正在寻找一种方法来创建一个索引数组,其中包括对一些条目的操作。例如,我有一组点,我想提取第一个,第二个和第一个减去第二个点。
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])
我知道我可以在这里使用一个函数,但是,我想保持它的简单和快速。
hgncfbus1#
你所做的在语法上是错误的:你应该做:
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])
1条答案
按热度按时间hgncfbus1#
你所做的在语法上是错误的:
你应该做:
或