gpt4 book ai didi

python - 如何提取第一个元素在不同列表中的二维数组中的那些行?

转载 作者:太空宇宙 更新时间:2023-11-04 10:35:04 25 4
gpt4 key购买 nike

<分区>

假设我有这个数组

array([[100,   1],
[200, 2],
[300, 3],
[400, 4],
[440, 3]])

我有这个列表或一维数组 [100,300]。我希望我的操作输出 [1,3]。我怎样才能在 numpy 中做到这一点。

我实际上是在 Theano(一个使用 gpu 加速计算的机器学习库)中使用这些 numpy 数组。我会有很多行。 Numpy 数组让我可以无缝地将它们用作 Theano 中的 Tensor 对象。但是如果我必须使用字典,我将不得不用普通的 Python 来做,而且我不确定一旦我转向大数据,这是否会很好。所以我实际上是在寻找一个 numpy 操作,一些索引技巧或类似的东西。

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com