我已经成功地使用numpy.array组合了两个列表(A,B),并将其分配给了一个新变量。第一个列表包含字符串(A),第二个列表包含整数(B)。当我打印新变量(C)时,我得到:
[[Str, Str]
[Int, Int]]
如果str =='XYZ',我现在试图查找对应的整数。如何使用索引函数在“ XYZ”旁边的行中找到int?
我最初的想法是将新列表分配给可变参数,即D:
D = C[A == 'XYZ',]
但是,我只会得到True,False等
对不起,我知道这很基本。
答案 0 :(得分:2)
这是您要完成的事情吗?
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])
答案 1 :(得分:0)
这是我想出的:
import numpy as np
A = ['ABC', 'XYZ']
B = [1, 99]
C = B[A == 'ABC']
D = B[A != 'ABC']
print(C)
print(B)