如何使用二维数组每一行的元素压缩一维数组中的每个元素？

Question

我正在尝试使用matplotlib绘制数据集，其中每个x包含多个y坐标。 要绘制它们，我必须组合这些数组以在单个图中显示它们。 如何使用来自二维数组每一行的元素（对应于一维数组元素的索引）压缩一维数组中的每个元素？ 不使用显式 for 循环。 使用内置函数（例如 zip/list comprehension）甚至更好： numpy ？

转向：

x = [1, 2, 3]
y = [[4, 5], [6, 7], [8, 9]]

进入：

r = [(1, 4), (1, 5), (2, 6), (2, 7), (3, 8), (3, 9)]

Answer 1

我想到了以下几点：

1. 将x扩展为与y相同的大小

2. 压平y

3. 压缩x和y

如下

>>> x = [1, 2, 3]
>>> y = [[4, 5], [6, 7], [8, 9]]
>>> x = np.array(x).repeat(2)
>>> y = np.array(y).reshape(-1)
>>> list(zip(x, y))
[(1, 4), (1, 5), (2, 6), (2, 7), (3, 8), (3, 9)]

我很想知道如何更有效地做到这一点。

请评论或回答更有效的方法。

Answer 2

x = [1, 2, 3]
y = [[4, 5], [6, 7], [8, 9]]

result = [(i[0], i[1][0]) for i in zip(x,y)] + [(i[0], i[1][1]) for i in zip(x,y)]

输出：

[(1, 4), (2, 6), (3, 8), (1, 5), (2, 7), (3, 9)]

Answer 3

通过方法5启发这里：

from functools import reduce

x = [1, 2, 3]
y = [[4, 5], [6, 7], [8, 9]]

def listOfTuples(l1, l2): 
    l = list(map(lambda x, y:[(x,y[0]), (x,y[1])], l1, l2))
    m = reduce(lambda x, y: x + y, l)
    return(m)

r = listOfTuples(x, y)

给

[(1, 4), (1, 5), (2, 6), (2, 7), (3, 8), (3, 9)]

Answer 4

repeat和reshape方法的变体：

In [89]: x = [1, 2, 3] 
    ...: y = [[4, 5], [6, 7], [8, 9]]                                                          
In [90]: res = np.zeros((3,2,2),int)                                                           
In [91]: res[:,:,1]=y                                                                          
In [92]: res[:,:,0]=np.array(x)[:,None]                                                        
In [93]: res                                                                                   
Out[93]: 
array([[[1, 4],
        [1, 5]],

       [[2, 6],
        [2, 7]],

       [[3, 8],
        [3, 9]]])
In [94]: res.reshape(6,2)                                                                      
Out[94]: 
array([[1, 4],
       [1, 5],
       [2, 6],
       [2, 7],
       [3, 8],
       [3, 9]])

它使用broadcasting将x和y映射到res数组。

Answer 5

这仅使用迭代器，使用itertools.chain.from_iterable

import itertools

x = [1, 2, 3]
y = [[4, 5], [6, 7], [8, 9]]

list(zip(list(itertools.chain.from_iterable(itertools.repeat(n, 2) for n in x)),
     list(itertools.chain.from_iterable(y))))

输出

[(1, 4), (1, 5), (2, 6), (2, 7), (3, 8), (3, 9)]

恕我直言，这在长期运行中应该更快。