我有一个字符串列表,如下所示:
l = [['apple','banana','kiwi'],['chair','table','spoon']]
给定一个字符串,我想要它在l中的索引。使用numpy进行实验,这就是我最终得到的结果:
import numpy as np
l = [['apple','banana','kiwi'],['chair','table','spoon']]
def ind(s):
i = [i for i in range(len(l)) if np.argwhere(np.array(l[i]) == s)][0]
j = np.argwhere(np.array(l[i]) == s)[0][0]
return i, j
s = ['apple','banana','kiwi','chair','table','spoon']
for val in s:
try:
print val, ind(val)
except IndexError:
print 'oops'
这对于苹果和椅子来说是失败的,它得到了一个索引器。而且,这对我来说很糟糕。有什么更好的方法来做这个吗?
如果你想使用numpy,你不需要自己滚动:
返回一个元组列表,该列表包含(外部列表索引、内部列表索引),设计为使您要查找的项可以位于多个内部列表中:
相关问题 更多 >
编程相关推荐