回答此问题可获得 20 贡献值,回答如果被采纳可获得 50 分。
<p>我有两个数组<code>a</code>和<code>b</code>,其中包含一些数据。我还有一个数组<code>select</code>,我想用它从a或b中进行选择。我只是想知道是否有一个pythonic方法可以这样做。下面是我当前的实现,它将a和b的每一行放入一个列表中,然后从中进行选择。你知道吗</p>
<pre><code>a = np.zeros(shape=(10,2,1,3,4))
b = np.ones(shape=(10,2,1,3,4))
select = [1,1,1,0,1,0,1,0,1,0]
c = []
for a1,b1,select1 in zip(a,b,select):
a1b1 = [a1,b1]
c.append(a1b1[select])
</code></pre>