numpy 按索引组合数组和列表元素

oknrviil  于 2022-11-10  发布在  其他
关注(0)|答案(2)|浏览(140)

我有一个数组(来自kdtree):

array =  [[a b c d e]
          [a b c d e]
          [a b c d e]]

还有一份名单:

lst = [1, 2, 3, 4, 5]

我想做一些列表理解(使用arraylst),使它看起来像这样:

desired_result = [[a, b, c, d, e, 1]
                  [a, b, c, d, e, 2]
                  [a, b, c, d, e, 3]]

我对列表理解很熟悉,只是还不够熟悉,不知道如何处理这个问题。

7fhtutme

7fhtutme1#

如果你想要一个列表理解:

result = [l+[x] for l,x in zip(array, lst)]
wwwo4jvm

wwwo4jvm2#

如果arraynumpy.ndarray类型,则列出理解的另一种方法:

result = numpy.c_[array, lst]

有关更多详细信息,请查看this answer

相关问题