python查找指定元素
三种方式对指定元素实现定位
-
本人也是新手,需要实现什么功能很多时候就是谷歌搜,一搜搜好几个才能解决问题,就像本文这个问题,我搜索到很多结果就只能找到第一个元素的索引,根本就文不是我想要的,反正大佬八成也看不到这篇文章,就随便写写啦
-
本文使用python自带功能、numpy库、pytorch库三种方式实现元素定位
-
写本文一为了方便比我还嫩的新手,二做个记录,所以很有可能我的写法繁琐了,如果您有更好的实现方法,欢迎指正,感谢!
数据准备
import numpy as np
import torch
element = 3 # 要找的元素
list_test = [1, 2, 2, 2, 3, 3, 4, 5, 6, 7, 3] # 一维数组
list_test_2 = [[1, 2, 3, 4, 3, 4, 6, 1], [1, 3, 2, 2, 3