python - 我想从列表中获取字符串

Question

有一个像下面这样的矩阵

[[a1, b, c],
[d2, e, a1],
[c1, dfd, u],
....

]

如果与特定单词匹配，则从矩阵返回索引。并找到行固定列。然后我将矩阵转换为列表。我得到这样的结果["RESULT DATA"]。我只想在RESULT DATA没有[''].

r, c = np.where(Matrix == SpecificWord)
find_row = np.where(c == 0) 
result_data = Matrix[find_row,1].tolist()
print(str(result_data ))
print(result_data[0])

标签： pythonpython-3.xlistnumpy

Answer 1

解决了

row, col = np.where(Terms == ElemData)
cList = col .tolist()
rLits = row.tolist()
if 0 not in cList:
    return None
else:
    num_col = cList.index(0)
    num_row = rLits[num_col]
    data = Terms[num_row, 1].tolist()
    return data