Home / Questions / 39070
Accepted
Jose Hermosilla Rodrigo
Asked: 2020-12-14 15:56:23 +0800 CST 2020-12-14 15:56:23 +0800 CST

Как вставить строки в массив numpy?

  • 772

У меня есть массив, созданный с помощью numpy:

matriz = numpy.matrix([]);

Я пытаюсь вставить строки в цикле (что-то вроде):

for i in range(3) : matriz = numpy.insert(matriz, i, [i + 1, 0, 0])

Я ожидал, что результат будет:

matrix([[1, 0, 0], [2, 0, 0], [3, 0, 0]])

Однако я получаю:

matrix([[ 1.,  2.,  3.,  0.,  0.,  0.,  0.,  0.,  0.]])

Что я делаю не так?

python

1 Answers

  1. Best Answer

    Это происходит из-за того, что вы не указываете ось, на которую хотите добавить данные ( 0для filasи 1для columnas). Если этот параметр не указан, массив/матрица сначала сглаживается , как вы можете видеть в документации numpy.insert():

    ось: целое, необязательно

    Ось, вдоль которой вставляются значения. Если ось имеет значение None, то arr сначала сглаживается.

    Правильный синтаксис будет:

    matriz=numpy.insert(matriz, i, [i + 1, 0, 0], axis=0)

    Проблема в том, что вы пытаетесь добавить строку с тремя столбцами в матрицу 1x0. Это выдает ошибку. Решение может заключаться в инициализации трехстрочного, но пустого массива. Для этого мы можем использовать numpy.empty:

    import numpy

matriz =  numpy.matrix(numpy.empty(shape=(0,3), dtype=int))
for i in range(3):
    matriz=numpy.insert(matriz, i, [i + 1, 0, 0], axis = 0)

    В любом случае есть более эффективные способы сделать это, например:

    import numpy

matriz = numpy.matrix([[i+1, 0, 0] for i in range(3)], dtype=int)

    Я указываю тип данных с помощью dtype, потому что если я этого не сделаю, он будет использоваться по floatумолчанию.

    В обоих случаях получаем:

    matrix([[1, 0, 0],    
        [2, 0, 0],    
        [3, 0, 0]])

    Хотя первый метод (с использованием np.insert()) гораздо менее эффективен, чем второй (списковые включения), я получаю следующие результаты для добавления 100 000 строк:

    np.insert(): 9,337375402450562 секунды
    list comprehensions: 0,07504963874816895 секунд

    • 2