我有25个随机数,希望每五个为一组把它们作为向量与给定的向量计算夹角
如图所示,cossim是计算夹角的函数,a为数组保存那25个数,我执行后返回计算结果最大的那一组,但是我并不希望返回一组数,而是希望给这组数一个编号后返回。例如25个数分为了ABCDE 5组,A组计算结果最大,就返回字符A,而不是[1,2,3,4,5]
应该如何在返回时给这组数一个名称,而不是只打印出这组数呢
你题目的解答代码如下:
n = 5
m = 0
z = 0
li = ['A','B','C','D','E']
for k, b in zip(li, [a[i:i+n] for i in range(0, len(a), n)]):
re = cossim(t, b)
if re > m:
m = re
z = k
print(z)
如有帮助,望采纳!谢谢!
你已经知道你要的结果在第几组了,维护一个内容为str1 = 'ABCDE'
的字符串,返回组号对应的字符即可
例如找到的值在第三组中,就返回str1[2]
,即C