numpy返回一个数组。我只需索引

时间:2017-07-12 22:34:52

标签: python arrays numpy

我有一个numpy数组:k = np.array([100,20,25,10,1,2])我试图使用np作为index=np.where(k<10),这给了我 index (array([4, 5]),)。我对某些事情感兴趣,只给我索引,所以我想index[0]=4而不是index[0]=[4 5]

我在numpy docs上找不到任何东西。

2 个答案:

答案 0 :(得分:2)

您可以获取结果中的第一个元素,如下所示:

index=np.where(k<10)[0]

然后indexarray([4, 5], dtype=int64),您可以根据需要访问index[0]index[1]

答案 1 :(得分:1)

你可以使用numpy.flatnonzero,它返回一个索引数组而不是数组的元组:

<td>@Html.ActionLink("Sprawdź","Details","Home",null,  new { @item.Id, @class = "btn btn-info" })</td>