numpy 忽略特定的行模式来展平列表

des4xlb0  于 2023-06-29  发布在  其他
关注(0)|答案(2)|浏览(70)

我有一个列表(65,2)如下:

[[1  2]
 [2  3]
 [3 4]
 [4 5]
 [5 6]]

现在我想把这个列表扁平化为(65),其中偶数行将被忽略。所需的输出应该是这样的。[1, 2, 3 , 4 , 5 , 6](第1行、第3行、第5行)

qnyhuwrf

qnyhuwrf1#

切片后展平:

>>> ar = np.column_stack([np.arange(1, 6), np.arange(2, 7)])
>>> ar
array([[1, 2],
       [2, 3],
       [3, 4],
       [4, 5],
       [5, 6]])
>>> ar[::2].ravel()
array([1, 2, 3, 4, 5, 6])
9vw9lbht

9vw9lbht2#

使用您的示例:

lists = [[1, 2],
         [2, 3],
         [3, 4],
         [4, 5],
         [5, 6]]

[x for pair in lists[::2] for x in pair]
# [1, 2, 3, 4, 5, 6]

相关问题