如何在二维数组中找到“ str”并在下一列中返回元素？

Question

我已经成功地使用numpy.array组合了两个列表（A，B），并将其分配给了一个新变量。第一个列表包含字符串（A），第二个列表包含整数（B）。当我打印新变量（C）时，我得到：

    [[Str, Str]
     [Int, Int]]

如果str =='XYZ'，我现在试图查找对应的整数。如何使用索引函数在“ XYZ”旁边的行中找到int？

我最初的想法是将新列表分配给可变参数，即D：

  D = C[A == 'XYZ',]

但是，我只会得到True，False等

对不起，我知道这很基本。

Answer 1

这是您要完成的事情吗？

import numpy as np my_matrix = np.array([['ABC','XYZ'],[1,99]]) print(my_matrix) my_index = my_matrix[0]=='XYZ' print(my_matrix[1][my_index])

Answer 2

这是我想出的：

  import numpy as np 
  A = ['ABC', 'XYZ']
  B = [1, 99]
  C = B[A == 'ABC']
  D = B[A != 'ABC']
  print(C)
  print(B)