python - 如何通过搜索匹配值来打印二维数组的特定元素

Question

假设我有一个二维数组:

a = np.array[ ("words",3) , ("blah",7) , ("hmm",1 ]      

我有一个值:3.0

如何根据匹配值打印二维数组中的第一个索引？

所需输出:单词

这是我的代码:

k = np.array([["words words ", 4], ["blah blah", 1], [" please help me", 9]])
a = []
for i in range(len(k)):
    holder = (int(k[i,1]))
    a = np.append(a, holder)
    print(k[i,1])
    print(a, 'a')

a = np.asarray(a)
print('')
print( a )
print('')

close = (np.abs(a - dot)).argmin()  # dot is the value 0, it is the value of previous dot product computation
print(close , "close")
c = close
print(a[c])

Answer 1

如果您只对打印带有特定数字的字符串感兴趣，我会这样做:

a = [("words",3), ("blah",7), ("hmm",1)]
b = [c[0] for c in a if c[1]==3]
print(b)
>>> b
['words']

如果您有兴趣寻找尽可能接近给定值的值，我会这样做:

1.构建一个数组b，其中包含a中存在的所有数值

2.使用b和dot中的值及其索引之间的绝对差值创建一个元组数组

3.找到最后一个数组中的最小值，并在b中获取相应的索引

4.b[index] 是要在 a 中查找的值。

def search_element(dot):
    b = [c[1] for c in a]
    index = min([(abs(f-dot),i) for i,f in enumerate(b)])[1]
    n = b[index]
    b = [c[0] for c in a if c[1]==n]
    return b