使用循环在两个数组或列表中查找匹配值
finding matching value in two arrays or lists using loop
我的语法如下:
a = []
data = ['2007-01-03', '2007-01-04', '2007-01-05','2007-01-03', '2007-01-04', '2007-01-05']
data1 = ['2007-01-04', '2007-01-05']
data = np.array(data)
data1 = np.array(data1)
for idx, day in enumerate(data):
for idx1, day1 in enumerate(data1):
print(np.where(day == day1)[0])
结果如下:
[]
[0]
[]
[]
[0]
[]
我应该如何修改语法以获得以下结果?
[1, 2, 4, 5]
res = [key for key, val in enumerate(data) if val in data1]
print(res)
输出:
[1, 2, 4, 5]
我的语法如下:
a = []
data = ['2007-01-03', '2007-01-04', '2007-01-05','2007-01-03', '2007-01-04', '2007-01-05']
data1 = ['2007-01-04', '2007-01-05']
data = np.array(data)
data1 = np.array(data1)
for idx, day in enumerate(data):
for idx1, day1 in enumerate(data1):
print(np.where(day == day1)[0])
结果如下:
[]
[0]
[]
[]
[0]
[]
我应该如何修改语法以获得以下结果?
[1, 2, 4, 5]
res = [key for key, val in enumerate(data) if val in data1]
print(res)
输出:
[1, 2, 4, 5]