Python屏蔽1d数组

时间:2017-07-25 15:32:51

标签: python arrays numpy boolean

说我有2个numpy数组

a = [0, 2, 4, 6]

b = [0.03, 0.78, 0.25, 0.47, 0.98, 0.58, 0.63]

我想返回一个第3个数组,它检查数组a的索引在b中的位置,并返回b中的索引值,如下所示:

c = [0.3, 0.25, 0.98, 0.63]

我试过了

for i in range(len(b)):
if b[i] == a.any():
    c=[I]

但得到所有0。

2 个答案:

答案 0 :(得分:5)

只需使用

c = b[a]

这将是b的视图,因此如果你改变c,你也可以改变b。如果您不想这样做,请使用.copy()

c = b[a].copy()

答案 1 :(得分:1)

使用列表理解

@app.route('/post' methods=['POST'])
相关问题