Início / coding / Perguntas / 79335463
Accepted
user28311778
Asked: 2025-01-07 17:28:28 +0800 CST

Como obter a matriz NumPy que desejo

  • 772

Como posso alterar a matriz dessa forma?

arr = [
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11],
[20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31],
[30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41],
...,
[130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141],
]

=> [
[ [0, 1], [20, 21], [30, 31], ,.., [130, 131]],
[ [2, 3], [22, 23], [32, 33], .., [132, 133]],
[ [4, 5], [24, 25], [34, 35], .., [134, 135]],
[ [6, 7], [26, 27], [36, 37], .., [136, 137]],
[ [8, 9], [28, 29], [38, 39], .., [138, 139]],
[ [10, 11], [30, 31], [40, 41], .., [140, 141]],
...
]

Minha tentativa foi melhor modificada para o seguinte, o que não foi o resultado que eu queria.

[ [0, 1], [2, 3], [4, 5], ,.., [10, 11]],
[ [20, 21], [22, 23], [24, 25], .., [30, 31]],
numpy

4 respostas

  1. Best Answer

    Você pode usar reshape+ swapaxes:

    n, m = arr.shape
out = arr.reshape(n, m//2, 2).swapaxes(0, 1)

# or
out = arr.reshape(n, -1, 2).swapaxes(0, 1)

    Exemplo:

    # input
arr = np.array([[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11],
                [20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31],
                [30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41],
                [130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141]])

# output
array([[[  0,   1], [ 20,  21], [ 30,  31], [130, 131]],
       [[  2,   3], [ 22,  23], [ 32,  33], [132, 133]],
       [[  4,   5], [ 24,  25], [ 34,  35], [134, 135]],
       [[  6,   7], [ 26,  27], [ 36,  37], [136, 137]],
       [[  8,   9], [ 28,  29], [ 38,  39], [138, 139]],
       [[ 10,  11], [ 30,  31], [ 40,  41], [140, 141]]])
    • 6

  2. Além da solução @Mozway, você pode usar stride para um uso mais avançado (e inseguro) do numpy:

    n, m = arr.shape
stride_x, stride_y = arr.strides
out = np.lib.stride_tricks.as_strided( arr, shape=(m//2,n,2), strides=(stride_y*2, stride_x, stride_y))

    Veja https://numpy.org/doc/stable/reference/generated/numpy.lib.stride_tricks.as_strided.html para documentação

    • 2

  3. Você quer fazer pareamento por colunas com passos de 2 (para pares). Imagine arrque você tem linhas e colunas. Você precisa iterar pelas colunas da lista original em pares e agrupar os elementos correspondentes de todas as linhas.

    result = [
    [[arr[row][col], arr[row][col + 1]] for row in range(len(arr))]
    for col in range(0, len(arr[0]), 2)
]
    • 0

  4. Você deve usar um for-loop duplo: um para percorrer os elementos em etapas de dois, e o segundo for-loop para criar pares de dois.

    output = [
    # Going over the elements with steps of 2.
    [
        # Create pairs of two
        [row[i], row[i + 1]]
        for row in arr
    ]
    for i in range(0, len(arr[0]), 2)
]

    Isto pressupõe, naturalmente, que todos os elementos de arrtenham o mesmo comprimento

    • 0

relate perguntas