在2D數組中查找值的索引

Question

這是我的代碼：

test_list= [
    ["Romeo and Juliet","Shakespeare"],
    ["Othello","Play"],
    ["Macbeth","Tragedy"]
]

value = "Tragedy"

print(test_list.index(value))

結果我得到“ ValueError：'悲劇'不在列表中

我得出的結論是.index僅適用於一維數組？ 但是那我該如何使2D陣列消失呢？ 如果我將數組設為1D，則此代碼可以正常工作。 請幫助，但簡單來說，我是個初學者。

很抱歉在手機上格式化問題。 該陣列為我正確設置了。

Answer 1

遍歷列表，並在每個子列表中搜索字符串。

Testlist = [
               ["Romeo and Juliet","Shakespeare"],
               ["Othello","Play"],
               ["Macbeth","Tragedy"]
               ]

Value = "Tragedy"

for index, lst in enumerate(Testlist):
  if Value in lst:
    print( index, lst.index(Value) )

Answer 2

您還可以使用地圖運算符：

# Get a boolean array - true if sublist contained the lookup value
value_in_sublist = map(lambda x: value in x, test_list)

# Get the index of the first True
print(value_in_sublist.index(True))

Answer 3

您也可以使用numpy ：

import numpy as np
test_list = np.array(test_list)
value = 'Tragedy'
print(np.where(test_list == value))

輸出：

(array([2]), array([1]))

如果一個元素有多次出現，則np.where將為您列出所有出現的索引。

Answer 4

numpy數組可能會在您的特定情況下有所幫助

import numpy

test_array = numpy.array(Testlist)
value = "Tragedy"

numpy.where(test_array==value)
# you will get (array([2]), array([1]))