在numpy

Question

如何使用numpy单独访问每行中的一个特定元素？

In[308]: cards
Out[296]: 
array([[  3.,   8.,   7.,  12.,   1.,   4.,  12.],
       [  5.,   6.,   2.,  11.,  10.,   9.,   6.],
       [  3.,   4.,   3.,   9.,   3.,   3.,  10.]])

以下内容将访问所有行中相同的元素[1,2,1]。但我想要第一行中的1行，第二行中的2行和第三行中的1行。

cards[:,[1,2,1]]
array([[ 8.,  7.,  8.],
       [ 6.,  2.,  6.],
       [ 4.,  3.,  4.]])

期望的输出：

array([[ 8.],
       [ 2.],
       [ 4.]])

Answer 1

您可以传递行和列的索引：

image_id

如果索引具有匹配的形状，则它们将成对处理。有关详情，请参阅documentation。

Answer 2

您可以在两个可迭代中传递单个元素：

mysqli_fetch_assoc