Início / coding / Perguntas / 79023711
Accepted
Ishigami
Asked: 2024-09-25 23:25:34 +0800 CST

Lista Python com matrizes NumPy como elementos

  • 772

Tenho uma lista com matrizes NumPy como elementos que se parece com isto:

[array([ 0.2, -2.3,  5.3]),
 array([-1.6, -1.7,  0.3]),
 array([ 2.4, -0.2, -3.0]),
 array([-4.1, -2.3, -2.7])]

e eu quero convertê-lo em 3 listas, cada uma com elementos das colunas da lista acima. Então o resultado desejado parece

list1 = [0.2, -1.6, 2.4, -4.1]
list2 = [-2.3, -1.7, -0.2, -2.3]
list3 = [5.3, 0.3, -3.0, -2.7]
python

7 respostas

  1. Best Answer

    Você pode conseguir isso usando o método NumPytransposeT como este:

    transposed = np.array(array_list).T
list1, list2, list3 = transposed.tolist()

    Exemplo completo

    • 6

  2. Você pode usar zip

    a = [
    np.array([0.2, -2.3, 5.3]),
    np.array([-1.6, -1.7, 0.3]),
    np.array([2.4, -0.2, -3.0]),
    np.array([-4.1, -2.3, -2.7])
]

    Com lista comp em uma linha usando zip:

    list1,list2,list3 = [list(x) for x in zip(*a)]
    • 4

  3. você pode tentar

    import numpy as np

a = [
    np.array([0.2, -2.3, 5.3]),
    np.array([-1.6, -1.7, 0.3]),
    np.array([2.4, -0.2, -3.0]),
    np.array([-4.1, -2.3, -2.7])
]

list1 = [arr[0] for arr in a]
list2 = [arr[1] for arr in a]
list3 = [arr[2] for arr in a]

# your result
print("list1:", list1)
print("list2:", list2)
print("list3:", list3)
    • 1

  4. Supondo que sua lista de matrizes foi definida da seguinte maneira:

    import numpy as np

array_list = [np.array([ 0.2, -2.3,  5.3]),
              np.array([-1.6, -1.7,  0.3]),
              np.array([ 2.4, -0.2, -3.0]),
              np.array([-4.1, -2.3, -2.7])]

    Então você pode usar a seguinte frase:

    list1, list2, list3 = np.transpose(array_list).tolist()

    A função NumPy transpose()converterá implicitamente array_listem um array NumPy bidimensional (então não há necessidade de fazer isso explicitamente) antes de transpô-lo (ou seja, transformar linhas em colunas e colunas em linhas). Chamar tolist()o array resultante o transformará em uma lista de listas, com três elementos (correspondentes às três colunas originais) no nível mais alto. Atribuir o resultado com list1, list2, list3 = ..."descompactará" essa lista de três elementos nas três variáveis ​​de destino.

    Se você concordar que os resultados ( list1, list2, list3) sejam novamente matrizes NumPy em vez de listas, você pode omitir tolist(), escrevendo apenas:

    list1, list2, list3 = np.transpose(array_list)
    • 1

  5. Você pode empilhar a lista em uma matriz numpy 2D e transpô-la.

    import numpy as np


array_list = [np.array([0.2, -2.3, 5.3]),
              np.array([-1.6, -1.7, 0.3]),
              np.array([2.4, -0.2, -3.0]),
              np.array([-4.1, -2.3, -2.7])]


stacked_array = np.vstack(array_list).T


list1, list2, list3 = stacked_array[0].tolist(), stacked_array[1].tolist(), stacked_array[2].tolist()

print(list1) print(list2) print(list3)
    • 0

  6. Seria mais apropriado converter a lista inteira em uma lista de listas 2D do Python em vez de instanciar três novas variáveis, de qualquer forma você pode fazer isso com

    original_list = [np.array([ 0.2, -2.3,  5.3]),
 np.array([-1.6, -1.7,  0.3]),
 np.array([ 2.4, -0.2, -3.0]),
 np.array([-4.1, -2.3, -2.7])]
 
list1 = [float(x[0]) for x in original_list]
list2 = [float(x[1]) for x in original_list]
list3 = [float(x[2]) for x in original_list]

    Ou você pode converter a lista original de listas em uma matriz numpy, transpô-la e convertê-la novamente em uma lista.

    result = np.array(original_list).transpose().tolist()

list1 = result[0]
list2 = result[1]
list3 = result[2]
    • -1 
  7. import numpy as np

# Given list of numpy arrays
arrays = [np.array([ 0.2, -2.3,  5.3]),
          np.array([-1.6, -1.7,  0.3]),
          np.array([ 2.4, -0.2, -3.0]),
          np.array([-4.1, -2.3, -2.7])]

# Transpose the arrays using zip and convert the result to lists
list1, list2, list3 = [list(x) for x in zip(*arrays)]

print("list1 =", list1)
print("list2 =", list2)
print("list3 =", list3)

'''Output:
list1 = [0.2, -1.6, 2.4, -4.1]
list2 = [-2.3, -1.7, -0.2, -2.3]
list3 = [5.3, 0.3, -3.0, -2.7]
'''
    • -1

relate perguntas