python – 从矩阵到单词列表

发布时间：2020-12-16 21:32:51 所属栏目：Python 来源：网络整理

导读：我有一个NumPy r by c矩阵的零和一.我有一个c字列表.我想返回一个长度为r的列表,其中每个元素是一个以空格分隔的字符串,仅由那些与该矩阵行中的1匹配的单词组成.这是一个例子： matrix=np.array([[0,1],[1,[0,1,1]])words=['python','c++','.net']output=[]f

我有一个NumPy r by c矩阵的零和一.我有一个c字列表.我想返回一个长度为r的列表,其中每个元素是一个以空格分隔的字符串,仅由那些与该矩阵行中的1匹配的单词组成.这是一个例子：

matrix=np.array([[0,1],[1,[0,1,1]])
words=['python','c++','.net']

output=[]
for row in range(matrix.shape[0]):
    output.append( ' '.join([words[i]  for i in range(matrix.shape[1]) if matrix[row,i]==1]))

什么是Pythonic完成此任务的方法？

谢谢,

解决方法

看吧：

>>> [' '.join(word for include_word,word in zip(row,words) if include_word) 
     for row in matrix]
['.net','python .net','c++ .net']

这是一个有趣的=).

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!