2d numpy.array（）将一个字符串与所有其他字符串进行比较，并对每个字符串重复

Question

假设我有一个带有字符串列表的输入文件（我在这里使用int表示整洁度），例如1,2,3,4,5,6，...，n

我想生成一个二维的numpy.array，看起来像这样：

a = [1,1], [1,2], [1,3], [1,4], ..., [1,n]

然后对每个连续的字符串重复，例如

a = [2,1], [2,2], [2,3], [2,4], ..., [2,n], ...[n,n]

我该怎么办？

Answer 1

查看itertools库。该product功能似乎是您所追求的In [19]: list( product([1, 2, 3, 4, 5], repeat=2) )Out[19]: [(1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (2, 1), (2, 2), (2, 3), (2, 4), (2, 5), (3, 1), (3, 2), (3, 3), (3, 4), (3, 5), (4, 1), (4, 2), (4, 3), (4, 4), (4, 5), (5, 1), (5, 2), (5, 3), (5, 4), (5, 5)]