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

oknrviil 于 2022-11-10 发布在其他

关注(0)|答案(2)|浏览(141)

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

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

还有一份名单：

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

我想做一些列表理解(使用array和lst)，使它看起来像这样：

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

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

2条答案

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

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

如果array是numpy.ndarray类型，则列出理解的另一种方法：

result = numpy.c_[array, lst]

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