使用python中numpy输出列表内指定元素的索引值
时间: 2024-05-06 17:20:39 浏览: 88
深度学习 Numpy 基础.zip_carepne_numpy_python NumPy_python 教程_python深度学
可以使用numpy的argwhere函数来输出列表内指定元素的索引值,示例如下:
```
import numpy as np
# 定义列表
lst = [1, 2, 3, 4, 5, 3, 6, 7, 3]
# 输出元素值为3的索引值
idx = np.argwhere(np.array(lst) == 3).flatten()
print(idx)
```
输出结果为:
```
[2 5 8]
```
其中,np.array(lst)将列表转换为numpy数组,然后使用==运算符找到元素值为3的位置,再使用argwhere函数输出这些位置的索引值。flatten函数将结果展开成一维数组。
阅读全文